LICENSE
README.md
pyproject.toml
assets/version.txt
custom_torchtitan.egg-info/PKG-INFO
custom_torchtitan.egg-info/SOURCES.txt
custom_torchtitan.egg-info/dependency_links.txt
custom_torchtitan.egg-info/requires.txt
custom_torchtitan.egg-info/top_level.txt
torchtitan/__init__.py
torchtitan/config_manager.py
torchtitan/train.py
torchtitan/components/checkpoint.py
torchtitan/components/dataloader.py
torchtitan/components/float8.py
torchtitan/components/ft.py
torchtitan/components/loss.py
torchtitan/components/optimizer.py
torchtitan/components/tokenizer.py
torchtitan/datasets/hf_datasets.py
torchtitan/datasets/tokenizer/__init__.py
torchtitan/datasets/tokenizer/tiktoken.py
torchtitan/distributed/__init__.py
torchtitan/distributed/parallel_dims.py
torchtitan/distributed/pipeline.py
torchtitan/distributed/utils.py
torchtitan/experiments/deepseek_v3/attn_mask_utils.py
torchtitan/experiments/deepseek_v3/checkpoint.py
torchtitan/experiments/deepseek_v3/download.py
torchtitan/experiments/deepseek_v3/model.py
torchtitan/experiments/deepseek_v3/model_config.py
torchtitan/experiments/deepseek_v3/run.py
torchtitan/experiments/deepseek_v3/symm_mem_recipes/__init__.py
torchtitan/experiments/deepseek_v3/symm_mem_recipes/triton_barrier.py
torchtitan/experiments/deepseek_v3/symm_mem_recipes/triton_on_device_all_to_all_v.py
torchtitan/experiments/deepseek_v3/symm_mem_recipes/triton_utils.py
torchtitan/models/__init__.py
torchtitan/models/norms.py
torchtitan/models/llama/__init__.py
torchtitan/models/llama/model.py
torchtitan/models/llama/parallelize_llama.py
torchtitan/models/llama/pipeline_llama.py
torchtitan/models/llama_multimodal/__init__.py
torchtitan/models/llama_multimodal/model.py
torchtitan/protocols/model_converter.py
torchtitan/protocols/train_spec.py
torchtitan/tools/logging.py
torchtitan/tools/metrics.py
torchtitan/tools/profiling.py
torchtitan/tools/utils.py