LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
src/accelerate/__init__.py
src/accelerate/accelerator.py
src/accelerate/big_modeling.py
src/accelerate/checkpointing.py
src/accelerate/data_loader.py
src/accelerate/hooks.py
src/accelerate/inference.py
src/accelerate/launchers.py
src/accelerate/local_sgd.py
src/accelerate/logging.py
src/accelerate/memory_utils.py
src/accelerate/optimizer.py
src/accelerate/scheduler.py
src/accelerate/state.py
src/accelerate/tracking.py
src/accelerate.egg-info/PKG-INFO
src/accelerate.egg-info/SOURCES.txt
src/accelerate.egg-info/dependency_links.txt
src/accelerate.egg-info/entry_points.txt
src/accelerate.egg-info/requires.txt
src/accelerate.egg-info/top_level.txt
src/accelerate/commands/__init__.py
src/accelerate/commands/accelerate_cli.py
src/accelerate/commands/env.py
src/accelerate/commands/estimate.py
src/accelerate/commands/launch.py
src/accelerate/commands/test.py
src/accelerate/commands/tpu.py
src/accelerate/commands/config/__init__.py
src/accelerate/commands/config/cluster.py
src/accelerate/commands/config/config.py
src/accelerate/commands/config/config_args.py
src/accelerate/commands/config/config_utils.py
src/accelerate/commands/config/default.py
src/accelerate/commands/config/sagemaker.py
src/accelerate/commands/config/update.py
src/accelerate/commands/menu/__init__.py
src/accelerate/commands/menu/cursor.py
src/accelerate/commands/menu/helpers.py
src/accelerate/commands/menu/input.py
src/accelerate/commands/menu/keymap.py
src/accelerate/commands/menu/selection_menu.py
src/accelerate/test_utils/__init__.py
src/accelerate/test_utils/examples.py
src/accelerate/test_utils/testing.py
src/accelerate/test_utils/training.py
src/accelerate/test_utils/scripts/__init__.py
src/accelerate/test_utils/scripts/test_cli.py
src/accelerate/test_utils/scripts/test_distributed_data_loop.py
src/accelerate/test_utils/scripts/test_notebook.py
src/accelerate/test_utils/scripts/test_ops.py
src/accelerate/test_utils/scripts/test_script.py
src/accelerate/test_utils/scripts/test_sync.py
src/accelerate/test_utils/scripts/external_deps/__init__.py
src/accelerate/test_utils/scripts/external_deps/test_checkpointing.py
src/accelerate/test_utils/scripts/external_deps/test_metrics.py
src/accelerate/test_utils/scripts/external_deps/test_peak_memory_usage.py
src/accelerate/test_utils/scripts/external_deps/test_performance.py
src/accelerate/test_utils/scripts/external_deps/test_pippy.py
src/accelerate/utils/__init__.py
src/accelerate/utils/bnb.py
src/accelerate/utils/constants.py
src/accelerate/utils/dataclasses.py
src/accelerate/utils/deepspeed.py
src/accelerate/utils/environment.py
src/accelerate/utils/fsdp_utils.py
src/accelerate/utils/imports.py
src/accelerate/utils/launch.py
src/accelerate/utils/megatron_lm.py
src/accelerate/utils/memory.py
src/accelerate/utils/modeling.py
src/accelerate/utils/offload.py
src/accelerate/utils/operations.py
src/accelerate/utils/other.py
src/accelerate/utils/random.py
src/accelerate/utils/rich.py
src/accelerate/utils/torch_xla.py
src/accelerate/utils/tqdm.py
src/accelerate/utils/transformer_engine.py
src/accelerate/utils/versions.py
tests/test_accelerator.py
tests/test_big_modeling.py
tests/test_cli.py
tests/test_cpu.py
tests/test_data_loader.py
tests/test_examples.py
tests/test_grad_sync.py
tests/test_hooks.py
tests/test_kwargs_handlers.py
tests/test_memory_utils.py
tests/test_metrics.py
tests/test_modeling_utils.py
tests/test_multigpu.py
tests/test_offload.py
tests/test_optimizer.py
tests/test_quantization.py
tests/test_sagemaker.py
tests/test_scheduler.py
tests/test_state_checkpointing.py
tests/test_tpu.py
tests/test_tracking.py
tests/test_utils.py