LICENSE.txt
setup.cfg
setup.py
sapiens_accelerator/__init__.py
sapiens_accelerator/accelerator.py
sapiens_accelerator/big_modeling.py
sapiens_accelerator/checkpointing.py
sapiens_accelerator/data_loader.py
sapiens_accelerator/hooks.py
sapiens_accelerator/inference.py
sapiens_accelerator/launchers.py
sapiens_accelerator/local_sgd.py
sapiens_accelerator/logging.py
sapiens_accelerator/memory_utils.py
sapiens_accelerator/optimizer.py
sapiens_accelerator/scheduler.py
sapiens_accelerator/state.py
sapiens_accelerator/tracking.py
sapiens_accelerator.egg-info/PKG-INFO
sapiens_accelerator.egg-info/SOURCES.txt
sapiens_accelerator.egg-info/dependency_links.txt
sapiens_accelerator.egg-info/entry_points.txt
sapiens_accelerator.egg-info/requires.txt
sapiens_accelerator.egg-info/top_level.txt
sapiens_accelerator/commands/__init__.py
sapiens_accelerator/commands/accelerate_cli.py
sapiens_accelerator/commands/env.py
sapiens_accelerator/commands/estimate.py
sapiens_accelerator/commands/launch.py
sapiens_accelerator/commands/merge.py
sapiens_accelerator/commands/test.py
sapiens_accelerator/commands/tpu.py
sapiens_accelerator/commands/utils.py
sapiens_accelerator/commands/config/__init__.py
sapiens_accelerator/commands/config/cluster.py
sapiens_accelerator/commands/config/config.py
sapiens_accelerator/commands/config/config_args.py
sapiens_accelerator/commands/config/config_utils.py
sapiens_accelerator/commands/config/default.py
sapiens_accelerator/commands/config/sagemaker.py
sapiens_accelerator/commands/config/update.py
sapiens_accelerator/commands/menu/__init__.py
sapiens_accelerator/commands/menu/cursor.py
sapiens_accelerator/commands/menu/helpers.py
sapiens_accelerator/commands/menu/input.py
sapiens_accelerator/commands/menu/keymap.py
sapiens_accelerator/commands/menu/selection_menu.py
sapiens_accelerator/test_utils/__init__.py
sapiens_accelerator/test_utils/examples.py
sapiens_accelerator/test_utils/testing.py
sapiens_accelerator/test_utils/training.py
sapiens_accelerator/test_utils/scripts/__init__.py
sapiens_accelerator/test_utils/scripts/test_cli.py
sapiens_accelerator/test_utils/scripts/test_ddp_comm_hook.py
sapiens_accelerator/test_utils/scripts/test_distributed_data_loop.py
sapiens_accelerator/test_utils/scripts/test_merge_weights.py
sapiens_accelerator/test_utils/scripts/test_notebook.py
sapiens_accelerator/test_utils/scripts/test_ops.py
sapiens_accelerator/test_utils/scripts/test_script.py
sapiens_accelerator/test_utils/scripts/test_sync.py
sapiens_accelerator/test_utils/scripts/external_deps/__init__.py
sapiens_accelerator/test_utils/scripts/external_deps/test_checkpointing.py
sapiens_accelerator/test_utils/scripts/external_deps/test_ds_multiple_model.py
sapiens_accelerator/test_utils/scripts/external_deps/test_metrics.py
sapiens_accelerator/test_utils/scripts/external_deps/test_peak_memory_usage.py
sapiens_accelerator/test_utils/scripts/external_deps/test_performance.py
sapiens_accelerator/test_utils/scripts/external_deps/test_pippy.py
sapiens_accelerator/test_utils/scripts/external_deps/test_zero3_integration.py
sapiens_accelerator/utils/__init__.py
sapiens_accelerator/utils/constants.py
sapiens_accelerator/utils/dataclasses.py
sapiens_accelerator/utils/deepspeed.py
sapiens_accelerator/utils/environment.py
sapiens_accelerator/utils/fsdp_utils.py
sapiens_accelerator/utils/imports.py
sapiens_accelerator/utils/launch.py
sapiens_accelerator/utils/megatron_lm.py
sapiens_accelerator/utils/memory.py
sapiens_accelerator/utils/modeling.py
sapiens_accelerator/utils/offload.py
sapiens_accelerator/utils/operations.py
sapiens_accelerator/utils/other.py
sapiens_accelerator/utils/random.py
sapiens_accelerator/utils/rich.py
sapiens_accelerator/utils/sapiens.py
sapiens_accelerator/utils/torch_xla.py
sapiens_accelerator/utils/tqdm.py
sapiens_accelerator/utils/transformer_engine.py
sapiens_accelerator/utils/versions.py