bridge.training

Subpackages

• bridge.training.tokenizers
  • bridge.training.tokenizers.tokenizer
  • bridge.training.tokenizers.config
• 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.mlflow_utils
  • bridge.training.utils.log_utils
  • bridge.training.utils.wandb_utils
  • bridge.training.utils.train_utils
  • bridge.training.utils.padding_utils
  • bridge.training.utils.checkpoint_utils
  • bridge.training.utils.comet_utils
  • bridge.training.utils.theoretical_memory_utils
  • bridge.training.utils.flop_utils
  • bridge.training.utils.batch_utils
  • bridge.training.utils.pg_utils
  • bridge.training.utils.visual_inputs
  • bridge.training.utils.packed_seq_utils
  • bridge.training.utils.config_utils
  • bridge.training.utils.omegaconf_utils
  • bridge.training.utils.sig_utils
  • bridge.training.utils.moe_token_drop
• bridge.training.post_training
  • bridge.training.post_training.distillation
  • bridge.training.post_training.checkpointing

Submodules

• bridge.training.llava_step
• bridge.training.setup
• bridge.training.tensor_inspect
• bridge.training.flex_dispatcher_backend
• bridge.training.train_megatron_mimo
• bridge.training.callbacks
• bridge.training.pretrain
• bridge.training.comm_overlap
• bridge.training.pretrain_megatron_mimo
• bridge.training.megatron_mimo_parallel_utils
• bridge.training.nvrx_straggler
• bridge.training.config
• bridge.training.audio_lm_step
• bridge.training.state
• bridge.training.eval_context_parallel_rebinding
• bridge.training.forward_step_func_types
• bridge.training.losses
• bridge.training.megatron_mimo_step
• bridge.training.gpt_step
• bridge.training.checkpointing
• bridge.training.vlm_step
• bridge.training.mixed_precision
• bridge.training.setup_megatron_mimo
• bridge.training.inprocess_restart
• bridge.training.fault_tolerance
• bridge.training.model_load_save
• bridge.training.distill
• bridge.training.nemotron_omni_step
• bridge.training.eval
• bridge.training.optim
• bridge.training.train
• bridge.training.initialize
• bridge.training.profiling
• bridge.training.finetune