Skip to main content
Ctrl+K
Megatron Bridge - Home Megatron Bridge - Home

Megatron Bridge

  • GitHub
Megatron Bridge - Home Megatron Bridge - Home

Megatron Bridge
  • GitHub

Table of Contents

Guides

  • Parallelisms Guide
  • Performance
  • Performance Tuning Guide
  • Using Recipes
  • NeMo 2.0 to Megatron Bridge Migration Guide
  • Megatron-LM to Megatron Bridge Guide

Bridge with 🤗 Hugging Face

  • Get Started with 🤗 Hugging Face Conversion
  • Megatron Bridge Conversion Technical Details

Supported Models

  • Large Language Models
    • DeepSeek V2
    • DeepSeek V3
    • Gemma 2
    • Gemma 3
    • GLM 4.5
    • GPT OSS
    • Llama 3
    • Llama Nemotron
    • Mistral
    • Moonlight
    • Nemotron 3 Nano
    • Nemotron 3 Super
    • Nemotron H and Nemotron Nano v2
    • OLMoE
    • Qwen
  • Vision Language Models
    • Gemma 3 VL (Vision-Language)
    • GLM-4.5V
    • Ministral 3
    • Nemotron Nano V2 VL
    • Qwen2.5-VL
    • Qwen3-VL
    • Qwen 3.5

Training and Customization

  • Configuration Overview
  • Training Entry Points
  • Training Loop Configuration
  • Optimizer and Scheduler Configuration
  • Logging and Monitoring
  • Profiling
  • Checkpointing
  • Megatron FSDP
  • Resiliency
  • Mixed Precision Training
  • CUDA Graphs
  • Hybrid / Hierarchical Context Parallel
  • Communication Overlap
  • Attention Optimizations
  • Activation Recomputation
  • CPU Offloading
  • Parameter-Efficient Fine-Tuning (PEFT)
  • Packed Sequences
  • Multi-Token Prediction (MTP)
  • Knowledge Distillation
  • Pruning
  • Callbacks

Model Optimization with ModelOpt

  • Quantization

Development

  • Contribute a New Model to Megatron Bridge
  • Adapting Megatron Bridge in Reinforcement Learning Frameworks
  • Documentation Development
  • API Reference
    • bridge
      • bridge.diffusion
        • bridge.diffusion.common
        • bridge.diffusion.conversion
        • bridge.diffusion.recipes
        • bridge.diffusion.data
        • bridge.diffusion.base
        • bridge.diffusion.models
      • bridge.training
        • bridge.training.mlm_compat
        • bridge.training.tokenizers
        • bridge.training.utils
        • bridge.training.post_training
        • bridge.training.losses
        • bridge.training.callbacks
        • bridge.training.inprocess_restart
        • bridge.training.profiling
        • bridge.training.flex_dispatcher_backend
        • bridge.training.eval
        • bridge.training.model_load_save
        • bridge.training.comm_overlap
        • bridge.training.pretrain
        • bridge.training.distill
        • bridge.training.setup
        • bridge.training.state
        • bridge.training.llava_step
        • bridge.training.train
        • bridge.training.config
        • bridge.training.tensor_inspect
        • bridge.training.nvrx_straggler
        • bridge.training.vlm_step
        • bridge.training.initialize
        • bridge.training.optim
        • bridge.training.mixed_precision
        • bridge.training.finetune
        • bridge.training.gpt_step
        • bridge.training.forward_step_func_types
        • bridge.training.checkpointing
        • bridge.training.fault_tolerance
      • bridge.peft
        • bridge.peft.dora_layers
        • bridge.peft.lora
        • bridge.peft.base
        • bridge.peft.dora
        • bridge.peft.utils
        • bridge.peft.module_matcher
        • bridge.peft.adapter_wrapper
        • bridge.peft.lora_layers
        • bridge.peft.canonical_lora
        • bridge.peft.walk_utils
        • bridge.peft.recompute
      • bridge.utils
        • bridge.utils.vocab_utils
        • bridge.utils.common_utils
        • bridge.utils.yaml_utils
        • bridge.utils.activation_map
        • bridge.utils.import_utils
        • bridge.utils.fusions
        • bridge.utils.instantiate_utils
        • bridge.utils.decorators
        • bridge.utils.slurm_utils
      • bridge.recipes
        • bridge.recipes.qwen_vl
        • bridge.recipes.qwen
        • bridge.recipes.gpt
        • bridge.recipes.gemma
        • bridge.recipes.glm
        • bridge.recipes.deepseek
        • bridge.recipes.utils
        • bridge.recipes.olmoe
        • bridge.recipes.gemma3_vl
        • bridge.recipes.glm_vl
        • bridge.recipes.moonlight
        • bridge.recipes.nemotronh
        • bridge.recipes.nemotron_vl
        • bridge.recipes.gpt_oss
        • bridge.recipes.llama
        • bridge.recipes.ministral3
        • bridge.recipes.common
        • bridge.recipes.run_plugins
      • bridge.data
        • bridge.data.mimo
        • bridge.data.datasets
        • bridge.data.hf_processors
        • bridge.data.builders
        • bridge.data.vlm_datasets
        • bridge.data.iterator_utils
        • bridge.data.finetuning
        • bridge.data.utils
        • bridge.data.loaders
        • bridge.data.samplers
      • bridge.models
        • bridge.models.common
        • bridge.models.qwen_vl
        • bridge.models.mamba
        • bridge.models.mimo
        • bridge.models.qwen
        • bridge.models.gpt
        • bridge.models.gemma
        • bridge.models.gemma_vl
        • bridge.models.glm
        • bridge.models.deepseek
        • bridge.models.llama_nemotron
        • bridge.models.olmoe
        • bridge.models.conversion
        • bridge.models.nemotron
        • bridge.models.mistral
        • bridge.models.decorators
        • bridge.models.qwen_omni
        • bridge.models.glm_vl
        • bridge.models.sarvam
        • bridge.models.nemotronh
        • bridge.models.nemotron_vl
        • bridge.models.hf_pretrained
        • bridge.models.gpt_oss
        • bridge.models.kimi
        • bridge.models.llama
        • bridge.models.ministral3
        • bridge.models.model_provider
        • bridge.models.gpt_full_te_layer_autocast_spec
        • bridge.models.transformer_config
        • bridge.models.mla_provider
        • bridge.models.config
        • bridge.models.distillation_provider
        • bridge.models.gpt_provider
        • bridge.models.t5_provider
      • bridge.package_info

Releases

  • Software Component Versions
  • Changelog
  • Known Issues

Agent Skills

  • Agent Skills Reference
    • Parallelism Strategy Selection Skill
    • CUDA Graphs
    • TP / DP / PP Communication Overlap Skill
    • Megatron FSDP Skill
    • Packed Sequences & Long-Context Training
    • Sequence Packing Skill
    • Hybrid / Hierarchical Context Parallel Skill
    • MoE Expert-Parallel Overlap Skill
    • MoE Communication Overlap
    • Resiliency

Directory Readme Files

  • Megatron Bridge Documentation
  • Supported Models
  • Large Language Models
  • Vision Language Models
  • Releases
  • Training and Customization
  • Model Optimization
  • API Reference
  • bridge
  • bridge.diffusion
  • bridge.diffusion.models
  • bridge.diffusion.models.common

bridge.diffusion.models.common#

Common modules for diffusion models in DFM.

Submodules#

  • bridge.diffusion.models.common.normalization
  • bridge.diffusion.models.common.dit_attention
  • bridge.diffusion.models.common.dit_embeddings

Package Contents#

Data#

__all__

API#

bridge.diffusion.models.common.__all__#

[‘DiTCrossAttention’, ‘DiTCrossAttentionSubmodules’, ‘DiTSelfAttention’, ‘ParallelTimestepEmbedding’…

previous

bridge.diffusion.models

next

bridge.diffusion.models.common.normalization
On this page
  • Submodules
  • Package Contents
    • Data
    • API
      • __all__
NVIDIA NVIDIA
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA Corporation.