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

Bridge with 🤗 Hugging Face

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

Training and Customization

  • Configuration Overview
  • Training Entry Points
  • Training Loop Configuration
  • Optimizer and Scheduler Configuration
  • Logging and Monitoring
  • Profiling
  • Checkpointing
  • Resiliency
  • Mixed Precision Training
  • Communication Overlap
  • Attention Optimizations
  • Activation Recomputation
  • CPU Offloading
  • Parameter-Efficient Fine-Tuning (PEFT)
  • Packed Sequences

Development

  • Contribute a New Model to Megatron Bridge
  • Documentation Development
  • API Reference
    • bridge
      • bridge.peft
        • bridge.peft.dora_layers
        • bridge.peft.lora
        • bridge.peft.dora
        • bridge.peft.adapter_wrapper
        • bridge.peft.walk_utils
        • bridge.peft.base
        • bridge.peft.module_matcher
        • bridge.peft.lora_layers
        • bridge.peft.canonical_lora
        • bridge.peft.utils
      • bridge.data
        • bridge.data.datasets
        • bridge.data.builders
        • bridge.data.hf_processors
        • bridge.data.loaders
        • bridge.data.samplers
        • bridge.data.utils
      • bridge.models
        • bridge.models.conversion
        • bridge.models.qwen
        • bridge.models.gemma
        • bridge.models.nemotron
        • bridge.models.qwen_vl
        • bridge.models.llama
        • bridge.models.mamba
        • bridge.models.deepseek
        • bridge.models.decorators
        • bridge.models.nemotronh
        • bridge.models.hf_pretrained
        • bridge.models.gpt_full_te_layer_autocast_spec
        • bridge.models.model_provider
        • bridge.models.config
        • bridge.models.gpt_provider
        • bridge.models.transformer_config
        • bridge.models.t5_provider
      • bridge.recipes
        • bridge.recipes.qwen
        • bridge.recipes.gpt
        • bridge.recipes.llama
        • bridge.recipes.mamba
        • bridge.recipes.utils
        • bridge.recipes.deepseek
        • bridge.recipes.nemotronh
        • bridge.recipes.run_plugins
      • bridge.utils
        • bridge.utils.vocab_utils
        • bridge.utils.instantiate_utils
        • bridge.utils.common_utils
        • bridge.utils.fusions
        • bridge.utils.yaml_utils
        • bridge.utils.import_utils
        • bridge.utils.decorators
      • bridge.training
        • bridge.training.tokenizers
        • bridge.training.mlm_compat
        • bridge.training.utils
        • bridge.training.model_load_save
        • bridge.training.fault_tolerance
        • bridge.training.pretrain
        • bridge.training.optim
        • bridge.training.state
        • bridge.training.train
        • bridge.training.nvrx_straggler
        • bridge.training.losses
        • bridge.training.eval
        • bridge.training.setup
        • bridge.training.initialize
        • bridge.training.checkpointing
        • bridge.training.inprocess_restart
        • bridge.training.config
        • bridge.training.comm_overlap
        • bridge.training.mixed_precision
        • bridge.training.finetune
        • bridge.training.deepep
        • bridge.training.profiling
        • bridge.training.gpt_step
        • bridge.training.forward_step_func_types
      • bridge.package_info
  • API Reference
  • bridge
  • bridge.models
  • bridge.models.mamba

bridge.models.mamba#

Submodules#

  • bridge.models.mamba.mamba_provider

Package Contents#

Data#

__all__

API#

bridge.models.mamba.__all__#

[‘MambaModelProvider’, ‘MambaModelProvider1P3B’, ‘MambaModelProvider2P7B’, ‘MambaModelProvider130M’,…

previous

bridge.models.llama.llama4_utils

next

bridge.models.mamba.mamba_provider
On this page
  • Submodules
  • Package Contents
    • Data
    • API
      • __all__
NVIDIA NVIDIA
Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2025, NVIDIA Corporation.