API Reference

This page contains auto-generated API reference documentation [1].

• 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.datasets.sft
      • bridge.data.datasets.packed_sequence
      • bridge.data.datasets.packing_utils
      • bridge.data.datasets.utils
    • bridge.data.builders
      • bridge.data.builders.hf_dataset
      • bridge.data.builders.finetuning_dataset
    • bridge.data.hf_processors
      • bridge.data.hf_processors.squad
    • bridge.data.loaders
    • bridge.data.samplers
    • bridge.data.utils
  • bridge.models
    • bridge.models.conversion
      • bridge.models.conversion.auto_bridge
      • bridge.models.conversion.param_mapping
      • bridge.models.conversion.mapping_registry
      • bridge.models.conversion.utils
      • bridge.models.conversion.model_bridge
    • bridge.models.qwen
      • bridge.models.qwen.qwen3_bridge
      • bridge.models.qwen.qwen_provider
      • bridge.models.qwen.qwen3_moe_bridge
      • bridge.models.qwen.qwen2_bridge
    • bridge.models.gemma
      • bridge.models.gemma.modules
      • bridge.models.gemma.gemma2_provider
      • bridge.models.gemma.gemma_provider
      • bridge.models.gemma.gemma_bridge
      • bridge.models.gemma.gemma2_bridge
    • bridge.models.nemotron
      • bridge.models.nemotron.nemotron_bridge
      • bridge.models.nemotron.nemotron_provider
    • bridge.models.qwen_vl
      • bridge.models.qwen_vl.qwen_vl_provider
      • bridge.models.qwen_vl.qwen25_vl_bridge
      • bridge.models.qwen_vl.modeling_qwen25_vl
    • bridge.models.llama
      • bridge.models.llama.llama_bridge
      • bridge.models.llama.llama_provider
      • bridge.models.llama.llama4_utils
    • bridge.models.mamba
      • bridge.models.mamba.mamba_provider
    • bridge.models.deepseek
      • bridge.models.deepseek.deepseek_v3_bridge
      • bridge.models.deepseek.common
      • bridge.models.deepseek.deepseek_v2_bridge
      • bridge.models.deepseek.deepseek_provider
    • bridge.models.decorators
      • bridge.models.decorators.torchrun
      • bridge.models.decorators.dispatch
    • bridge.models.nemotronh
      • bridge.models.nemotronh.nemotron_h_bridge
      • bridge.models.nemotronh.nemotron_h_provider
    • bridge.models.hf_pretrained
      • bridge.models.hf_pretrained.causal_lm
      • bridge.models.hf_pretrained.state
      • bridge.models.hf_pretrained.base
      • bridge.models.hf_pretrained.safe_config_loader
      • bridge.models.hf_pretrained.vlm
    • 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.qwen.qwen3
      • bridge.recipes.qwen.qwen2
      • bridge.recipes.qwen.qwen3_moe
    • bridge.recipes.gpt
      • bridge.recipes.gpt.gpt3_175b
    • bridge.recipes.llama
      • bridge.recipes.llama.llama3
      • bridge.recipes.llama.llama2
    • bridge.recipes.mamba
      • bridge.recipes.mamba.mamba2_780m
      • bridge.recipes.mamba.mamba2_2_7b
      • bridge.recipes.mamba.mamba2_hybrid_8b
      • bridge.recipes.mamba.mamba2_1_3b
      • bridge.recipes.mamba.mamba2_130m
      • bridge.recipes.mamba.mamba2_8b
      • bridge.recipes.mamba.mamba2_370m
    • bridge.recipes.utils
      • bridge.recipes.utils.tokenizer_utils
      • bridge.recipes.utils.nemo_run_utils
      • bridge.recipes.utils.dataset_utils
      • bridge.recipes.utils.optimizer_utils
    • bridge.recipes.deepseek
      • bridge.recipes.deepseek.deepseek_v2_lite
      • bridge.recipes.deepseek.deepseek_v3
      • bridge.recipes.deepseek.deepseek_v2
    • bridge.recipes.nemotronh
      • bridge.recipes.nemotronh.nemotron_nano_12b_v2
      • bridge.recipes.nemotronh.nemotronh_8b
      • bridge.recipes.nemotronh.nemotron_nano_9b_v2
      • bridge.recipes.nemotronh.nemotronh_47b
      • bridge.recipes.nemotronh.nemotronh_56b
      • bridge.recipes.nemotronh.nemotronh_4b
    • 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.tokenizers.gpt2_tokenization
      • bridge.training.tokenizers.tokenizer
      • bridge.training.tokenizers.config
      • bridge.training.tokenizers.bert_tokenization
      • bridge.training.tokenizers.multimodal_tokenizer
    • bridge.training.mlm_compat
      • bridge.training.mlm_compat.model
      • bridge.training.mlm_compat.activations
      • bridge.training.mlm_compat.arguments
    • bridge.training.utils
      • bridge.training.utils.omegaconf_utils
      • bridge.training.utils.log_utils
      • bridge.training.utils.moe_token_drop
      • bridge.training.utils.train_utils
      • bridge.training.utils.flop_utils
      • bridge.training.utils.theoretical_memory_utils
      • bridge.training.utils.sig_utils
      • bridge.training.utils.wandb_utils
      • bridge.training.utils.config_utils
      • bridge.training.utils.checkpoint_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

[1]

Created with sphinx-autodoc2