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/_diffusers/__init__.py
nemo_automodel/_diffusers/_hf_cache.py
nemo_automodel/_diffusers/auto_diffusion_pipeline.py
nemo_automodel/_transformers/__init__.py
nemo_automodel/_transformers/auto_model.py
nemo_automodel/_transformers/auto_tokenizer.py
nemo_automodel/_transformers/capabilities.py
nemo_automodel/_transformers/infrastructure.py
nemo_automodel/_transformers/kernel_patches.py
nemo_automodel/_transformers/mfu.py
nemo_automodel/_transformers/model_capabilities.py
nemo_automodel/_transformers/model_init.py
nemo_automodel/_transformers/registry.py
nemo_automodel/_transformers/retrieval.py
nemo_automodel/_transformers/te_attention.py
nemo_automodel/_transformers/utils.py
nemo_automodel/_transformers/tokenization/nemo_auto_tokenizer.py
nemo_automodel/_transformers/tokenization/registry.py
nemo_automodel/_transformers/tokenization/tokenization_mistral_common.py
nemo_automodel/_transformers/v4_patches/__init__.py
nemo_automodel/_transformers/v4_patches/kv_sharing.py
nemo_automodel/_transformers/v4_patches/layer_types.py
nemo_automodel/_transformers/v4_patches/rotary.py
nemo_automodel/autonvtx/__init__.py
nemo_automodel/cli/__init__.py
nemo_automodel/cli/app.py
nemo_automodel/cli/query_capabilities.py
nemo_automodel/cli/utils.py
nemo_automodel/components/__init__.py
nemo_automodel/components/_peft/__init__.py
nemo_automodel/components/_peft/lora.py
nemo_automodel/components/_peft/lora_experts.py
nemo_automodel/components/_peft/lora_kernel.py
nemo_automodel/components/_peft/lora_mlp.py
nemo_automodel/components/_peft/module_matcher.py
nemo_automodel/components/attention/__init__.py
nemo_automodel/components/attention/dflash_mask.py
nemo_automodel/components/attention/flex_attention.py
nemo_automodel/components/attention/utils.py
nemo_automodel/components/checkpoint/__init__.py
nemo_automodel/components/checkpoint/_torch_backports.py
nemo_automodel/components/checkpoint/addons.py
nemo_automodel/components/checkpoint/checkpointing.py
nemo_automodel/components/checkpoint/config.py
nemo_automodel/components/checkpoint/conversion_mapping.py
nemo_automodel/components/checkpoint/state_dict_adapter.py
nemo_automodel/components/checkpoint/stateful_wrappers.py
nemo_automodel/components/checkpoint/utils.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/lazy_mapped_dataset.py
nemo_automodel/components/datasets/reservoir_sampler.py
nemo_automodel/components/datasets/utils.py
nemo_automodel/components/datasets/audio/__init__.py
nemo_automodel/components/datasets/audio/collate_fns.py
nemo_automodel/components/datasets/audio/datasets.py
nemo_automodel/components/datasets/audio/multi_en.py
nemo_automodel/components/datasets/diffusion/__init__.py
nemo_automodel/components/datasets/diffusion/base_dataset.py
nemo_automodel/components/datasets/diffusion/collate_fns.py
nemo_automodel/components/datasets/diffusion/meta_files_dataset.py
nemo_automodel/components/datasets/diffusion/mock_dataloader.py
nemo_automodel/components/datasets/diffusion/multi_tier_bucketing.py
nemo_automodel/components/datasets/diffusion/sampler.py
nemo_automodel/components/datasets/diffusion/text_to_image_dataset.py
nemo_automodel/components/datasets/diffusion/text_to_video_dataset.py
nemo_automodel/components/datasets/dllm/__init__.py
nemo_automodel/components/datasets/dllm/collate.py
nemo_automodel/components/datasets/dllm/corruption.py
nemo_automodel/components/datasets/llm/__init__.py
nemo_automodel/components/datasets/llm/agent_chat.py
nemo_automodel/components/datasets/llm/chat_dataset.py
nemo_automodel/components/datasets/llm/column_mapped_text_instruction_dataset.py
nemo_automodel/components/datasets/llm/column_mapped_text_instruction_iterable_dataset.py
nemo_automodel/components/datasets/llm/delta_lake_dataset.py
nemo_automodel/components/datasets/llm/eagle3.py
nemo_automodel/components/datasets/llm/eagle3_cache.py
nemo_automodel/components/datasets/llm/formatting_utils.py
nemo_automodel/components/datasets/llm/hellaswag.py
nemo_automodel/components/datasets/llm/length_grouped_sampler.py
nemo_automodel/components/datasets/llm/megatron_dataset.py
nemo_automodel/components/datasets/llm/mock.py
nemo_automodel/components/datasets/llm/mock_iterable_dataset.py
nemo_automodel/components/datasets/llm/mock_packed.py
nemo_automodel/components/datasets/llm/mock_seq_cls.py
nemo_automodel/components/datasets/llm/nanogpt_dataset.py
nemo_automodel/components/datasets/llm/neat_packing.py
nemo_automodel/components/datasets/llm/packed_sequence.py
nemo_automodel/components/datasets/llm/retrieval_collator.py
nemo_automodel/components/datasets/llm/retrieval_dataset.py
nemo_automodel/components/datasets/llm/retrieval_dataset_inline.py
nemo_automodel/components/datasets/llm/seq_cls.py
nemo_automodel/components/datasets/llm/squad.py
nemo_automodel/components/datasets/llm/xlam.py
nemo_automodel/components/datasets/llm/megatron/Makefile
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/multimodal/__init__.py
nemo_automodel/components/datasets/multimodal/collate_fns.py
nemo_automodel/components/datasets/multimodal/datasets.py
nemo_automodel/components/datasets/multimodal/distributed_iterable.py
nemo_automodel/components/datasets/multimodal/interleave.py
nemo_automodel/components/datasets/multimodal/packing.py
nemo_automodel/components/datasets/multimodal/parquet_utils.py
nemo_automodel/components/datasets/multimodal/transforms.py
nemo_automodel/components/datasets/multimodal/utils.py
nemo_automodel/components/datasets/multimodal/video.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/fake_image.py
nemo_automodel/components/datasets/vlm/mock.py
nemo_automodel/components/datasets/vlm/neat_packing_vlm.py
nemo_automodel/components/datasets/vlm/pp_media.py
nemo_automodel/components/datasets/vlm/samplers.py
nemo_automodel/components/datasets/vlm/utils.py
nemo_automodel/components/distributed/__init__.py
nemo_automodel/components/distributed/activation_checkpointing.py
nemo_automodel/components/distributed/config.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/mamba_cp.py
nemo_automodel/components/distributed/megatron_fsdp.py
nemo_automodel/components/distributed/mesh.py
nemo_automodel/components/distributed/mesh_utils.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/parallelizer_utils.py
nemo_automodel/components/distributed/tensor_utils.py
nemo_automodel/components/distributed/thd_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/config.py
nemo_automodel/components/distributed/pipelining/functional.py
nemo_automodel/components/distributed/pipelining/hf_utils.py
nemo_automodel/components/eval/__init__.py
nemo_automodel/components/eval/tool_call_evaluator.py
nemo_automodel/components/eval/tool_call_parser.py
nemo_automodel/components/flow_matching/__init__.py
nemo_automodel/components/flow_matching/pipeline.py
nemo_automodel/components/flow_matching/time_shift_utils.py
nemo_automodel/components/flow_matching/adapters/__init__.py
nemo_automodel/components/flow_matching/adapters/base.py
nemo_automodel/components/flow_matching/adapters/flux.py
nemo_automodel/components/flow_matching/adapters/flux2.py
nemo_automodel/components/flow_matching/adapters/hunyuan.py
nemo_automodel/components/flow_matching/adapters/qwen_image.py
nemo_automodel/components/flow_matching/adapters/simple.py
nemo_automodel/components/launcher/__init__.py
nemo_automodel/components/launcher/base.py
nemo_automodel/components/launcher/interactive.py
nemo_automodel/components/launcher/nemo_run/__init__.py
nemo_automodel/components/launcher/nemo_run/config.py
nemo_automodel/components/launcher/nemo_run/launcher.py
nemo_automodel/components/launcher/nemo_run/utils.py
nemo_automodel/components/launcher/skypilot/__init__.py
nemo_automodel/components/launcher/skypilot/config.py
nemo_automodel/components/launcher/skypilot/launcher.py
nemo_automodel/components/launcher/skypilot/utils.py
nemo_automodel/components/loggers/__init__.py
nemo_automodel/components/loggers/comet_utils.py
nemo_automodel/components/loggers/log_utils.py
nemo_automodel/components/loggers/loggers.py
nemo_automodel/components/loggers/metric_logger.py
nemo_automodel/components/loggers/mlflow_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/dllm_loss.py
nemo_automodel/components/loss/kd_loss.py
nemo_automodel/components/loss/linear_ce.py
nemo_automodel/components/loss/loss.py
nemo_automodel/components/loss/masked_ce.py
nemo_automodel/components/loss/mtp.py
nemo_automodel/components/loss/soft_ce.py
nemo_automodel/components/loss/te_parallel_ce.py
nemo_automodel/components/loss/utils.py
nemo_automodel/components/loss/benchmark/soft_ce_loss.py
nemo_automodel/components/loss/triton/soft_cross_entropy.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/bagel/__init__.py
nemo_automodel/components/models/bagel/attention_masks.py
nemo_automodel/components/models/bagel/autoencoder.py
nemo_automodel/components/models/bagel/configuration.py
nemo_automodel/components/models/bagel/connector.py
nemo_automodel/components/models/bagel/embeddings.py
nemo_automodel/components/models/bagel/hf_backbone_loader.py
nemo_automodel/components/models/bagel/model.py
nemo_automodel/components/models/bagel/modeling_qwen2_packed.py
nemo_automodel/components/models/bagel/modeling_siglip_navit.py
nemo_automodel/components/models/bagel/state_dict_adapter.py
nemo_automodel/components/models/baichuan/__init__.py
nemo_automodel/components/models/baichuan/configuration.py
nemo_automodel/components/models/baichuan/model.py
nemo_automodel/components/models/common/__init__.py
nemo_automodel/components/models/common/bidirectional.py
nemo_automodel/components/models/common/gated_delta_net_fp32.py
nemo_automodel/components/models/common/hf_checkpointing_mixin.py
nemo_automodel/components/models/common/inbatch_neg_utils.py
nemo_automodel/components/models/common/packing.py
nemo_automodel/components/models/common/utils.py
nemo_automodel/components/models/common/mtp/__init__.py
nemo_automodel/components/models/common/mtp/mtp.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/rope_utils.py
nemo_automodel/components/models/deepseek_v3/state_dict_adapter.py
nemo_automodel/components/models/deepseek_v32/__init__.py
nemo_automodel/components/models/deepseek_v32/config.py
nemo_automodel/components/models/deepseek_v32/layers.py
nemo_automodel/components/models/deepseek_v32/model.py
nemo_automodel/components/models/deepseek_v32/state_dict_adapter.py
nemo_automodel/components/models/deepseek_v4/__init__.py
nemo_automodel/components/models/deepseek_v4/config.py
nemo_automodel/components/models/deepseek_v4/fsdp.py
nemo_automodel/components/models/deepseek_v4/layers.py
nemo_automodel/components/models/deepseek_v4/model.py
nemo_automodel/components/models/deepseek_v4/mtp.py
nemo_automodel/components/models/deepseek_v4/optimized_kernels.py
nemo_automodel/components/models/deepseek_v4/state_dict_adapter.py
nemo_automodel/components/models/deepseek_v4/kernels/__init__.py
nemo_automodel/components/models/deepseek_v4/kernels/_tilelang.py
nemo_automodel/components/models/deepseek_v4/kernels/sparse_attention.py
nemo_automodel/components/models/deepseek_v4/kernels/tilelang_indexer.py
nemo_automodel/components/models/deepseek_v4/kernels/tilelang_indexer_bwd.py
nemo_automodel/components/models/deepseek_v4/kernels/tilelang_indexer_fwd.py
nemo_automodel/components/models/deepseek_v4/kernels/tilelang_sparse_mla_bwd.py
nemo_automodel/components/models/deepseek_v4/kernels/tilelang_sparse_mla_fwd.py
nemo_automodel/components/models/ernie4_5/__init__.py
nemo_automodel/components/models/ernie4_5/model.py
nemo_automodel/components/models/ernie4_5/rope_utils.py
nemo_automodel/components/models/ernie4_5/state_dict_adapter.py
nemo_automodel/components/models/gemma4_drafter/__init__.py
nemo_automodel/components/models/gemma4_drafter/composite.py
nemo_automodel/components/models/gemma4_drafter/model.py
nemo_automodel/components/models/gemma4_moe/__init__.py
nemo_automodel/components/models/gemma4_moe/model.py
nemo_automodel/components/models/gemma4_moe/state_dict_adapter.py
nemo_automodel/components/models/glm4_moe/__init__.py
nemo_automodel/components/models/glm4_moe/layers.py
nemo_automodel/components/models/glm4_moe/model.py
nemo_automodel/components/models/glm4_moe/state_dict_adapter.py
nemo_automodel/components/models/glm4_moe_lite/__init__.py
nemo_automodel/components/models/glm4_moe_lite/model.py
nemo_automodel/components/models/glm_moe_dsa/__init__.py
nemo_automodel/components/models/glm_moe_dsa/model.py
nemo_automodel/components/models/glm_moe_dsa/state_dict_adapter.py
nemo_automodel/components/models/gpt_oss/__init__.py
nemo_automodel/components/models/gpt_oss/layers.py
nemo_automodel/components/models/gpt_oss/model.py
nemo_automodel/components/models/gpt_oss/rope_utils.py
nemo_automodel/components/models/gpt_oss/state_dict_adapter.py
nemo_automodel/components/models/hy_mt2/__init__.py
nemo_automodel/components/models/hy_mt2/config.py
nemo_automodel/components/models/hy_mt2/dispatch.py
nemo_automodel/components/models/hy_mt2/layers.py
nemo_automodel/components/models/hy_mt2/model.py
nemo_automodel/components/models/hy_mt2/state_dict_adapter.py
nemo_automodel/components/models/hy_v3/__init__.py
nemo_automodel/components/models/hy_v3/config.py
nemo_automodel/components/models/hy_v3/layers.py
nemo_automodel/components/models/hy_v3/model.py
nemo_automodel/components/models/hy_v3/state_dict_adapter.py
nemo_automodel/components/models/kimi_k25_vl/__init__.py
nemo_automodel/components/models/kimi_k25_vl/model.py
nemo_automodel/components/models/kimi_k25_vl/state_dict_adapter.py
nemo_automodel/components/models/kimivl/__init__.py
nemo_automodel/components/models/kimivl/model.py
nemo_automodel/components/models/ling_v2/__init__.py
nemo_automodel/components/models/ling_v2/config.py
nemo_automodel/components/models/ling_v2/layers.py
nemo_automodel/components/models/ling_v2/model.py
nemo_automodel/components/models/ling_v2/state_dict_adapter.py
nemo_automodel/components/models/llama/__init__.py
nemo_automodel/components/models/llama/model.py
nemo_automodel/components/models/llama/rope_utils.py
nemo_automodel/components/models/llama/state_dict_adapter.py
nemo_automodel/components/models/llama_bidirectional/__init__.py
nemo_automodel/components/models/llama_bidirectional/export_onnx.py
nemo_automodel/components/models/llama_bidirectional/model.py
nemo_automodel/components/models/llava_onevision/__init__.py
nemo_automodel/components/models/llava_onevision/model.py
nemo_automodel/components/models/llava_onevision/rice_vit.py
nemo_automodel/components/models/llava_onevision/state_dict_adapter.py
nemo_automodel/components/models/mimo_v2_flash/__init__.py
nemo_automodel/components/models/mimo_v2_flash/config.py
nemo_automodel/components/models/mimo_v2_flash/model.py
nemo_automodel/components/models/mimo_v2_flash/state_dict_adapter.py
nemo_automodel/components/models/minimax_m2/__init__.py
nemo_automodel/components/models/minimax_m2/layers.py
nemo_automodel/components/models/minimax_m2/model.py
nemo_automodel/components/models/minimax_m2/state_dict_adapter.py
nemo_automodel/components/models/ministral_bidirectional/__init__.py
nemo_automodel/components/models/ministral_bidirectional/model.py
nemo_automodel/components/models/mistral3/__init__.py
nemo_automodel/components/models/mistral3/model.py
nemo_automodel/components/models/mistral3_vlm/__init__.py
nemo_automodel/components/models/mistral3_vlm/model.py
nemo_automodel/components/models/mistral3_vlm/state_dict_adapter.py
nemo_automodel/components/models/mistral4/__init__.py
nemo_automodel/components/models/mistral4/configuration.py
nemo_automodel/components/models/mistral4/model.py
nemo_automodel/components/models/mistral4/state_dict_adapter.py
nemo_automodel/components/models/nemotron_omni/__init__.py
nemo_automodel/components/models/nemotron_omni/model.py
nemo_automodel/components/models/nemotron_omni/state_dict_adapter.py
nemo_automodel/components/models/nemotron_parse/__init__.py
nemo_automodel/components/models/nemotron_parse/model.py
nemo_automodel/components/models/nemotron_parse/nemotron_parse_loss.py
nemo_automodel/components/models/nemotron_v3/__init__.py
nemo_automodel/components/models/nemotron_v3/cache.py
nemo_automodel/components/models/nemotron_v3/layers.py
nemo_automodel/components/models/nemotron_v3/model.py
nemo_automodel/components/models/nemotron_v3/mtp.py
nemo_automodel/components/models/nemotron_v3/state_dict_adapter.py
nemo_automodel/components/models/qwen2/__init__.py
nemo_automodel/components/models/qwen2/model.py
nemo_automodel/components/models/qwen2/state_dict_adapter.py
nemo_automodel/components/models/qwen2_5_omni/__init__.py
nemo_automodel/components/models/qwen2_5_omni/model.py
nemo_automodel/components/models/qwen2_5_omni/state_dict_adapter.py
nemo_automodel/components/models/qwen3_5/__init__.py
nemo_automodel/components/models/qwen3_5/model.py
nemo_automodel/components/models/qwen3_5/state_dict_adapter.py
nemo_automodel/components/models/qwen3_5_moe/__init__.py
nemo_automodel/components/models/qwen3_5_moe/cp_linear_attn.py
nemo_automodel/components/models/qwen3_5_moe/model.py
nemo_automodel/components/models/qwen3_5_moe/state_dict_adapter.py
nemo_automodel/components/models/qwen3_moe/__init__.py
nemo_automodel/components/models/qwen3_moe/layers.py
nemo_automodel/components/models/qwen3_moe/model.py
nemo_automodel/components/models/qwen3_moe/state_dict_adapter.py
nemo_automodel/components/models/qwen3_next/__init__.py
nemo_automodel/components/models/qwen3_next/layers.py
nemo_automodel/components/models/qwen3_next/model.py
nemo_automodel/components/models/qwen3_next/state_dict_adapter.py
nemo_automodel/components/models/qwen3_omni_moe/__init__.py
nemo_automodel/components/models/qwen3_omni_moe/model.py
nemo_automodel/components/models/qwen3_omni_moe/state_dict_adapter.py
nemo_automodel/components/models/qwen3_vl_moe/__init__.py
nemo_automodel/components/models/qwen3_vl_moe/model.py
nemo_automodel/components/models/qwen3_vl_moe/state_dict_adapter.py
nemo_automodel/components/models/step3p5/__init__.py
nemo_automodel/components/models/step3p5/layers.py
nemo_automodel/components/models/step3p5/model.py
nemo_automodel/components/models/step3p5/state_dict_adapter.py
nemo_automodel/components/models/step3p7/__init__.py
nemo_automodel/components/models/step3p7/configuration_step3p7.py
nemo_automodel/components/models/step3p7/model.py
nemo_automodel/components/models/step3p7/mtp.py
nemo_automodel/components/models/step3p7/processing_step3.py
nemo_automodel/components/models/step3p7/state_dict_adapter.py
nemo_automodel/components/models/step3p7/vision_encoder.py
nemo_automodel/components/moe/__init__.py
nemo_automodel/components/moe/config.py
nemo_automodel/components/moe/experts.py
nemo_automodel/components/moe/fsdp_mixin.py
nemo_automodel/components/moe/layers.py
nemo_automodel/components/moe/load_balance_metrics.py
nemo_automodel/components/moe/mxfp8.py
nemo_automodel/components/moe/parallelizer.py
nemo_automodel/components/moe/state_dict_mixin.py
nemo_automodel/components/moe/state_dict_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/moe/uccl_ep/__init__.py
nemo_automodel/components/moe/uccl_ep/_buffer.py
nemo_automodel/components/moe/uccl_ep/_utils.py
nemo_automodel/components/moe/uccl_ep/buffer.py
nemo_automodel/components/optim/__init__.py
nemo_automodel/components/optim/dion.py
nemo_automodel/components/optim/optimizer.py
nemo_automodel/components/optim/precision_warnings.py
nemo_automodel/components/optim/scheduler.py
nemo_automodel/components/quantization/__init__.py
nemo_automodel/components/quantization/fp8.py
nemo_automodel/components/quantization/qat.py
nemo_automodel/components/quantization/qlora.py
nemo_automodel/components/speculative/__init__.py
nemo_automodel/components/speculative/bench_sglang.py
nemo_automodel/components/speculative/precompute_eagle3.py
nemo_automodel/components/speculative/regenerate.py
nemo_automodel/components/speculative/serve_sglang.py
nemo_automodel/components/speculative/serve_target.py
nemo_automodel/components/speculative/dflash/__init__.py
nemo_automodel/components/speculative/dflash/core.py
nemo_automodel/components/speculative/dflash/draft_qwen3.py
nemo_automodel/components/speculative/dflash/registry.py
nemo_automodel/components/speculative/dflash/target.py
nemo_automodel/components/speculative/eagle/__init__.py
nemo_automodel/components/speculative/eagle/backend.py
nemo_automodel/components/speculative/eagle/core.py
nemo_automodel/components/speculative/eagle/core_v12.py
nemo_automodel/components/speculative/eagle/draft_gpt_oss.py
nemo_automodel/components/speculative/eagle/draft_llama.py
nemo_automodel/components/speculative/eagle/draft_llama_v12.py
nemo_automodel/components/speculative/eagle/peagle_attention.py
nemo_automodel/components/speculative/eagle/peagle_data.py
nemo_automodel/components/speculative/eagle/peagle_draft.py
nemo_automodel/components/speculative/eagle/peagle_trainer.py
nemo_automodel/components/speculative/eagle/registry.py
nemo_automodel/components/speculative/eagle/target.py
nemo_automodel/components/speculative/eagle/target_v12.py
nemo_automodel/components/speculative/eagle/remote/__init__.py
nemo_automodel/components/speculative/eagle/remote/client.py
nemo_automodel/components/speculative/eagle/remote/protocol.py
nemo_automodel/components/speculative/eagle/remote/server.py
nemo_automodel/components/speculative/eagle/remote/transport.py
nemo_automodel/components/speculative/eagle/remote/wire.py
nemo_automodel/components/training/__init__.py
nemo_automodel/components/training/ema.py
nemo_automodel/components/training/garbage_collection.py
nemo_automodel/components/training/model_output_utils.py
nemo_automodel/components/training/neftune.py
nemo_automodel/components/training/rng.py
nemo_automodel/components/training/signal_handler.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/flops_utils.py
nemo_automodel/components/utils/model_utils.py
nemo_automodel/components/utils/yaml_utils.py
nemo_automodel/recipes/_dist_utils.py
nemo_automodel/recipes/_typed_config.py
nemo_automodel/recipes/base_recipe.py
nemo_automodel/recipes/diffusion/__init__.py
nemo_automodel/recipes/diffusion/train.py
nemo_automodel/recipes/dllm/__init__.py
nemo_automodel/recipes/dllm/strategy.py
nemo_automodel/recipes/dllm/train_ft.py
nemo_automodel/recipes/llm/benchmark.py
nemo_automodel/recipes/llm/kd.py
nemo_automodel/recipes/llm/peagle_recipe.py
nemo_automodel/recipes/llm/train_dflash.py
nemo_automodel/recipes/llm/train_eagle1.py
nemo_automodel/recipes/llm/train_eagle2.py
nemo_automodel/recipes/llm/train_eagle3.py
nemo_automodel/recipes/llm/train_ft.py
nemo_automodel/recipes/llm/train_seq_cls.py
nemo_automodel/recipes/multimodal/__init__.py
nemo_automodel/recipes/multimodal/finetune.py
nemo_automodel/recipes/multimodal/pretrain.py
nemo_automodel/recipes/retrieval/__init__.py
nemo_automodel/recipes/retrieval/mine_hard_negatives.py
nemo_automodel/recipes/retrieval/train_bi_encoder.py
nemo_automodel/recipes/retrieval/train_cross_encoder.py
nemo_automodel/recipes/vlm/finetune.py
nemo_automodel/recipes/vlm/kd.py
nemo_automodel/shared/__init__.py
nemo_automodel/shared/import_utils.py
nemo_automodel/shared/te_patches.py
nemo_automodel/shared/torch_patches.py
nemo_automodel/shared/transformers_patches.py
nemo_automodel/shared/utils.py
tests/test_greedy_knapsack_perf.py
tests/test_media_token_consistency.py
tests/test_meta_dataset_all.py