LICENSE
README.md
pyproject.toml
setup.cfg
src/accmt/__init__.py
src/accmt/callbacks.py
src/accmt/collate_fns.py
src/accmt/curriculum.py
src/accmt/debug_timings.py
src/accmt/decorators.py
src/accmt/evaluator.py
src/accmt/hp_search.py
src/accmt/hyperparameters.py
src/accmt/metrics.py
src/accmt/model_wrapper.py
src/accmt/modules.py
src/accmt/monitor.py
src/accmt/samplers.py
src/accmt/states.py
src/accmt/tqdm.py
src/accmt/tracker.py
src/accmt/trainer.py
src/accmt/tunnel.py
src/accmt.egg-info/PKG-INFO
src/accmt.egg-info/SOURCES.txt
src/accmt.egg-info/dependency_links.txt
src/accmt.egg-info/entry_points.txt
src/accmt.egg-info/requires.txt
src/accmt.egg-info/top_level.txt
src/accmt/utils/__init__.py
src/accmt/utils/availability.py
src/accmt/utils/data.py
src/accmt/utils/distributed.py
src/accmt/utils/globals.py
src/accmt/utils/maps.py
src/accmt/utils/misc.py
src/accmt/utils/seed.py
src/accmt/utils/tensor.py
src/cli/__init__.py
src/cli/accmt.py
src/cli/launch.py
src/cli/parser.py
src/cli/utils.py
src/cli/config/ddp-bf16.yaml
src/cli/config/ddp-fp16.yaml
src/cli/config/ddp-fp32.yaml
src/cli/config/ddp-fp8.yaml
src/cli/config/ddp.yaml
src/cli/config/deepspeed-0-bf16.yaml
src/cli/config/deepspeed-0-fp16.yaml
src/cli/config/deepspeed-0-fp32.yaml
src/cli/config/deepspeed-0-fp8.yaml
src/cli/config/deepspeed-0.yaml
src/cli/config/deepspeed-1-bf16.yaml
src/cli/config/deepspeed-1-fp16.yaml
src/cli/config/deepspeed-1-fp32.yaml
src/cli/config/deepspeed-1-fp8.yaml
src/cli/config/deepspeed-1.yaml
src/cli/config/deepspeed-2-bf16-cpu-cpu.yaml
src/cli/config/deepspeed-2-bf16-cpu-none.yaml
src/cli/config/deepspeed-2-bf16-cpu-nvme.yaml
src/cli/config/deepspeed-2-bf16-none-cpu.yaml
src/cli/config/deepspeed-2-bf16-none-none.yaml
src/cli/config/deepspeed-2-bf16-none-nvme.yaml
src/cli/config/deepspeed-2-bf16-nvme-cpu.yaml
src/cli/config/deepspeed-2-bf16-nvme-none.yaml
src/cli/config/deepspeed-2-bf16-nvme-nvme.yaml
src/cli/config/deepspeed-2-bf16.yaml
src/cli/config/deepspeed-2-fp16-cpu-cpu.yaml
src/cli/config/deepspeed-2-fp16-cpu-none.yaml
src/cli/config/deepspeed-2-fp16-cpu-nvme.yaml
src/cli/config/deepspeed-2-fp16-none-cpu.yaml
src/cli/config/deepspeed-2-fp16-none-none.yaml
src/cli/config/deepspeed-2-fp16-none-nvme.yaml
src/cli/config/deepspeed-2-fp16-nvme-cpu.yaml
src/cli/config/deepspeed-2-fp16-nvme-none.yaml
src/cli/config/deepspeed-2-fp16-nvme-nvme.yaml
src/cli/config/deepspeed-2-fp16.yaml
src/cli/config/deepspeed-2-fp32-cpu-cpu.yaml
src/cli/config/deepspeed-2-fp32-cpu-none.yaml
src/cli/config/deepspeed-2-fp32-cpu-nvme.yaml
src/cli/config/deepspeed-2-fp32-none-cpu.yaml
src/cli/config/deepspeed-2-fp32-none-none.yaml
src/cli/config/deepspeed-2-fp32-none-nvme.yaml
src/cli/config/deepspeed-2-fp32-nvme-cpu.yaml
src/cli/config/deepspeed-2-fp32-nvme-none.yaml
src/cli/config/deepspeed-2-fp32-nvme-nvme.yaml
src/cli/config/deepspeed-2-fp32.yaml
src/cli/config/deepspeed-2-fp8-cpu-cpu.yaml
src/cli/config/deepspeed-2-fp8-cpu-none.yaml
src/cli/config/deepspeed-2-fp8-cpu-nvme.yaml
src/cli/config/deepspeed-2-fp8-none-cpu.yaml
src/cli/config/deepspeed-2-fp8-none-none.yaml
src/cli/config/deepspeed-2-fp8-none-nvme.yaml
src/cli/config/deepspeed-2-fp8-nvme-cpu.yaml
src/cli/config/deepspeed-2-fp8-nvme-none.yaml
src/cli/config/deepspeed-2-fp8-nvme-nvme.yaml
src/cli/config/deepspeed-2-fp8.yaml
src/cli/config/deepspeed-2.yaml
src/cli/config/deepspeed-3-bf16-cpu-cpu.yaml
src/cli/config/deepspeed-3-bf16-cpu-none.yaml
src/cli/config/deepspeed-3-bf16-cpu-nvme.yaml
src/cli/config/deepspeed-3-bf16-none-cpu.yaml
src/cli/config/deepspeed-3-bf16-none-none.yaml
src/cli/config/deepspeed-3-bf16-none-nvme.yaml
src/cli/config/deepspeed-3-bf16-nvme-cpu.yaml
src/cli/config/deepspeed-3-bf16-nvme-none.yaml
src/cli/config/deepspeed-3-bf16-nvme-nvme.yaml
src/cli/config/deepspeed-3-bf16.yaml
src/cli/config/deepspeed-3-fp16-cpu-cpu.yaml
src/cli/config/deepspeed-3-fp16-cpu-none.yaml
src/cli/config/deepspeed-3-fp16-cpu-nvme.yaml
src/cli/config/deepspeed-3-fp16-none-cpu.yaml
src/cli/config/deepspeed-3-fp16-none-none.yaml
src/cli/config/deepspeed-3-fp16-none-nvme.yaml
src/cli/config/deepspeed-3-fp16-nvme-cpu.yaml
src/cli/config/deepspeed-3-fp16-nvme-none.yaml
src/cli/config/deepspeed-3-fp16-nvme-nvme.yaml
src/cli/config/deepspeed-3-fp16.yaml
src/cli/config/deepspeed-3-fp32-cpu-cpu.yaml
src/cli/config/deepspeed-3-fp32-cpu-none.yaml
src/cli/config/deepspeed-3-fp32-cpu-nvme.yaml
src/cli/config/deepspeed-3-fp32-none-cpu.yaml
src/cli/config/deepspeed-3-fp32-none-none.yaml
src/cli/config/deepspeed-3-fp32-none-nvme.yaml
src/cli/config/deepspeed-3-fp32-nvme-cpu.yaml
src/cli/config/deepspeed-3-fp32-nvme-none.yaml
src/cli/config/deepspeed-3-fp32-nvme-nvme.yaml
src/cli/config/deepspeed-3-fp32.yaml
src/cli/config/deepspeed-3-fp8-cpu-cpu.yaml
src/cli/config/deepspeed-3-fp8-cpu-none.yaml
src/cli/config/deepspeed-3-fp8-cpu-nvme.yaml
src/cli/config/deepspeed-3-fp8-none-cpu.yaml
src/cli/config/deepspeed-3-fp8-none-none.yaml
src/cli/config/deepspeed-3-fp8-none-nvme.yaml
src/cli/config/deepspeed-3-fp8-nvme-cpu.yaml
src/cli/config/deepspeed-3-fp8-nvme-none.yaml
src/cli/config/deepspeed-3-fp8-nvme-nvme.yaml
src/cli/config/deepspeed-3-fp8.yaml
src/cli/config/deepspeed-3.yaml
src/cli/config/deepspeed.yaml
src/cli/config/fsdp-bf16-cpu.yaml
src/cli/config/fsdp-bf16.yaml
src/cli/config/fsdp-fp16-cpu.yaml
src/cli/config/fsdp-fp16.yaml
src/cli/config/fsdp-fp32-cpu.yaml
src/cli/config/fsdp-fp32.yaml
src/cli/config/fsdp-fp8-cpu.yaml
src/cli/config/fsdp-fp8.yaml
src/cli/config/fsdp.yaml
src/cli/example/hps_example.yaml
src/tests/__init__.py
src/tests/curriculum.py
src/tests/dummy_callbacks.py
src/tests/dummy_datasets.py
src/tests/dummy_metrics.py
src/tests/dummy_modules.py
src/tests/trainer.py
src/tests/utils.py