path	# lines of code
utils/log_reports.py	147
utils/stale.py	41
benchmarks/torch.compile/regional_compilation.py	51
benchmarks/fsdp2/measure_utils.py	82
benchmarks/fsdp2/main.py	81
benchmarks/fsdp2/visualize.py	80
benchmarks/fsdp2/utils.py	169
benchmarks/big_model_inference/big_model_inference.py	105
benchmarks/big_model_inference/measures_util.py	57
benchmarks/fp8/torchao/fp8_utils.py	66
benchmarks/fp8/torchao/non_distributed.py	100
benchmarks/fp8/torchao/fsdp.py	121
benchmarks/fp8/torchao/distrib_deepspeed.py	163
benchmarks/fp8/torchao/ddp.py	107
benchmarks/fp8/transformer_engine/fp8_utils.py	66
benchmarks/fp8/transformer_engine/non_distributed.py	87
benchmarks/fp8/transformer_engine/fsdp.py	110
benchmarks/fp8/transformer_engine/distrib_deepspeed.py	141
benchmarks/fp8/transformer_engine/ddp.py	95
benchmarks/fp8/ms_amp/fp8_utils.py	67
benchmarks/fp8/ms_amp/non_distributed.py	77
benchmarks/fp8/ms_amp/distrib_deepspeed.py	118
benchmarks/fp8/ms_amp/ddp.py	81
src/accelerate/scheduler.py	41
src/accelerate/checkpointing.py	208
src/accelerate/inference.py	112
src/accelerate/utils/rich.py	6
src/accelerate/utils/other.py	260
src/accelerate/utils/__init__.py	279
src/accelerate/utils/environment.py	201
src/accelerate/utils/imports.py	338
src/accelerate/utils/deepspeed.py	178
src/accelerate/utils/dataclasses.py	1815
src/accelerate/utils/bnb.py	309
src/accelerate/utils/operations.py	434
src/accelerate/utils/modeling.py	1371
src/accelerate/utils/transformer_engine.py	100
src/accelerate/utils/constants.py	82
src/accelerate/utils/fsdp_utils.py	489
src/accelerate/utils/torch_xla.py	18
src/accelerate/utils/random.py	114
src/accelerate/utils/memory.py	103
src/accelerate/utils/versions.py	14
src/accelerate/utils/tqdm.py	16
src/accelerate/utils/launch.py	593
src/accelerate/utils/megatron_lm.py	1032
src/accelerate/utils/offload.py	120
src/accelerate/utils/ao.py	43
src/accelerate/__init__.py	34
src/accelerate/accelerator.py	2103
src/accelerate/big_modeling.py	384
src/accelerate/memory_utils.py	6
src/accelerate/optimizer.py	135
src/accelerate/commands/to_fsdp2.py	118
src/accelerate/commands/accelerate_cli.py	27
src/accelerate/commands/__init__.py	1
src/accelerate/commands/config/cluster.py	809
src/accelerate/commands/config/__init__.py	21
src/accelerate/commands/config/update.py	33
src/accelerate/commands/config/sagemaker.py	233
src/accelerate/commands/config/default.py	124
src/accelerate/commands/config/config.py	56
src/accelerate/commands/config/config_args.py	195
src/accelerate/commands/config/config_utils.py	79
src/accelerate/commands/menu/__init__.py	1
src/accelerate/commands/menu/keymap.py	96
src/accelerate/commands/menu/input.py	40
src/accelerate/commands/menu/helpers.py	23
src/accelerate/commands/menu/selection_menu.py	108
src/accelerate/commands/menu/cursor.py	36
src/accelerate/commands/tpu.py	118
src/accelerate/commands/env.py	93
src/accelerate/commands/estimate.py	212
src/accelerate/commands/merge.py	39
src/accelerate/commands/utils.py	72
src/accelerate/commands/launch.py	1076
src/accelerate/hooks.py	437
src/accelerate/local_sgd.py	42
src/accelerate/data_loader.py	876
src/accelerate/launchers.py	176
src/accelerate/state.py	771
src/accelerate/tracking.py	572
src/accelerate/logging.py	40
pyproject.toml	38
setup.py	84
manim_animations/dataloaders/stage_6.py	164
manim_animations/dataloaders/stage_5.py	173
manim_animations/dataloaders/stage_3.py	19
manim_animations/dataloaders/stage_2.py	141
manim_animations/dataloaders/stage_7.py	154
manim_animations/dataloaders/stage_4.py	33
manim_animations/dataloaders/stage_0.py	16
manim_animations/dataloaders/stage_1.py	16
manim_animations/big_model_inference/stage_5.py	161
manim_animations/big_model_inference/stage_3.py	114
manim_animations/big_model_inference/stage_2.py	86
manim_animations/big_model_inference/stage_4.py	113
manim_animations/big_model_inference/stage_1.py	71