LICENSE
README.md
pyproject.toml
nemo_automodel/__init__.py
nemo_automodel/package_info.py
nemo_automodel.egg-info/PKG-INFO
nemo_automodel.egg-info/SOURCES.txt
nemo_automodel.egg-info/dependency_links.txt
nemo_automodel.egg-info/entry_points.txt
nemo_automodel.egg-info/requires.txt
nemo_automodel.egg-info/top_level.txt
nemo_automodel/_cli/app.py
nemo_automodel/components/__init__.py
nemo_automodel/components/_peft/__init__.py
nemo_automodel/components/_peft/lora.py
nemo_automodel/components/_peft/lora_kernel.py
nemo_automodel/components/_peft/module_matcher.py
nemo_automodel/components/_transformers/__init__.py
nemo_automodel/components/_transformers/auto_model.py
nemo_automodel/components/_transformers/utils.py
nemo_automodel/components/checkpoint/__init__.py
nemo_automodel/components/checkpoint/_torch_backports.py
nemo_automodel/components/checkpoint/checkpointing.py
nemo_automodel/components/checkpoint/state_dict_adapter.py
nemo_automodel/components/checkpoint/stateful_wrappers.py
nemo_automodel/components/checkpoint/_backports/__init__.py
nemo_automodel/components/checkpoint/_backports/_fsspec_filesystem.py
nemo_automodel/components/checkpoint/_backports/_version.py
nemo_automodel/components/checkpoint/_backports/consolidate_hf_safetensors.py
nemo_automodel/components/checkpoint/_backports/default_planner.py
nemo_automodel/components/checkpoint/_backports/filesystem.py
nemo_automodel/components/checkpoint/_backports/hf_storage.py
nemo_automodel/components/checkpoint/_backports/hf_utils.py
nemo_automodel/components/checkpoint/_backports/planner_helpers.py
nemo_automodel/components/config/__init__.py
nemo_automodel/components/config/_arg_parser.py
nemo_automodel/components/config/loader.py
nemo_automodel/components/datasets/__init__.py
nemo_automodel/components/datasets/utils.py
nemo_automodel/components/datasets/llm/__init__.py
nemo_automodel/components/datasets/llm/column_mapped_text_instruction_dataset.py
nemo_automodel/components/datasets/llm/formatting_utils.py
nemo_automodel/components/datasets/llm/hellaswag.py
nemo_automodel/components/datasets/llm/megatron_dataset.py
nemo_automodel/components/datasets/llm/mock.py
nemo_automodel/components/datasets/llm/mock_packed.py
nemo_automodel/components/datasets/llm/nanogpt_dataset.py
nemo_automodel/components/datasets/llm/packed_sequence.py
nemo_automodel/components/datasets/llm/squad.py
nemo_automodel/components/datasets/llm/megatron/__init__.py
nemo_automodel/components/datasets/llm/megatron/builder.py
nemo_automodel/components/datasets/llm/megatron/gpt_dataset.py
nemo_automodel/components/datasets/llm/megatron/helpers.py
nemo_automodel/components/datasets/llm/megatron/indexed_dataset.py
nemo_automodel/components/datasets/llm/megatron/megatron_utils.py
nemo_automodel/components/datasets/llm/megatron/sampler.py
nemo_automodel/components/datasets/vlm/__init__.py
nemo_automodel/components/datasets/vlm/collate_fns.py
nemo_automodel/components/datasets/vlm/datasets.py
nemo_automodel/components/datasets/vlm/utils.py
nemo_automodel/components/distributed/__init__.py
nemo_automodel/components/distributed/cp_utils.py
nemo_automodel/components/distributed/ddp.py
nemo_automodel/components/distributed/fsdp2.py
nemo_automodel/components/distributed/grad_utils.py
nemo_automodel/components/distributed/init_utils.py
nemo_automodel/components/distributed/megatron_fsdp.py
nemo_automodel/components/distributed/optimized_tp_plans.py
nemo_automodel/components/distributed/parallel_styles.py
nemo_automodel/components/distributed/parallelizer.py
nemo_automodel/components/distributed/tensor_utils.py
nemo_automodel/components/distributed/utils.py
nemo_automodel/components/distributed/pipelining/__init__.py
nemo_automodel/components/distributed/pipelining/autopipeline.py
nemo_automodel/components/distributed/pipelining/functional.py
nemo_automodel/components/distributed/pipelining/hf_utils.py
nemo_automodel/components/launcher/__init__.py
nemo_automodel/components/launcher/slurm/config.py
nemo_automodel/components/launcher/slurm/template.py
nemo_automodel/components/launcher/slurm/utils.py
nemo_automodel/components/loggers/__init__.py
nemo_automodel/components/loggers/log_utils.py
nemo_automodel/components/loggers/wandb_utils.py
nemo_automodel/components/loss/__init__.py
nemo_automodel/components/loss/chunked_ce.py
nemo_automodel/components/loss/kd_loss.py
nemo_automodel/components/loss/linear_ce.py
nemo_automodel/components/loss/masked_ce.py
nemo_automodel/components/loss/te_parallel_ce.py
nemo_automodel/components/loss/triton/te_cross_entropy.py
nemo_automodel/components/models/__init__.py
nemo_automodel/components/models/gpt2.py
nemo_automodel/components/models/deepseek_v3/__init__.py
nemo_automodel/components/models/deepseek_v3/layers.py
nemo_automodel/components/models/deepseek_v3/model.py
nemo_automodel/components/models/deepseek_v3/parallelizer.py
nemo_automodel/components/models/deepseek_v3/state_dict_adapter.py
nemo_automodel/components/moe/__init__.py
nemo_automodel/components/moe/layers.py
nemo_automodel/components/moe/rope_utils.py
nemo_automodel/components/moe/state_dict_mixin.py
nemo_automodel/components/moe/state_dict_utils.py
nemo_automodel/components/moe/utils.py
nemo_automodel/components/moe/megatron/__init__.py
nemo_automodel/components/moe/megatron/fused_a2a.py
nemo_automodel/components/moe/megatron/fused_indices_converter.py
nemo_automodel/components/moe/megatron/moe_utils.py
nemo_automodel/components/moe/megatron/token_dispatcher.py
nemo_automodel/components/optim/__init__.py
nemo_automodel/components/optim/scheduler.py
nemo_automodel/components/quantization/__init__.py
nemo_automodel/components/quantization/fp8.py
nemo_automodel/components/quantization/qlora.py
nemo_automodel/components/training/__init__.py
nemo_automodel/components/training/rng.py
nemo_automodel/components/training/step_scheduler.py
nemo_automodel/components/training/timers.py
nemo_automodel/components/training/utils.py
nemo_automodel/components/utils/__init__.py
nemo_automodel/components/utils/compile_utils.py
nemo_automodel/components/utils/model_utils.py
nemo_automodel/components/utils/sig_utils.py
nemo_automodel/components/utils/yaml_utils.py
nemo_automodel/recipes/base_recipe.py
nemo_automodel/recipes/llm/kd.py
nemo_automodel/recipes/llm/train_ft.py
nemo_automodel/recipes/vlm/finetune.py
nemo_automodel/shared/__init__.py
nemo_automodel/shared/import_utils.py
nemo_automodel/shared/utils.py