colossalai.context

• colossalai.context.process_group_initializer
  • Initializer_Tensor
  • Initializer_Sequence
  • Initializer_Pipeline
  • Initializer_Data
  • Initializer_2p5D
  • Initializer_2D
  • Initializer_3D
  • Initializer_1D
  • ProcessGroupInitializer
  • Initializer_Model
  • colossalai.context.process_group_initializer.initializer_1d
  • colossalai.context.process_group_initializer.initializer_2d
  • colossalai.context.process_group_initializer.initializer_2p5d
  • colossalai.context.process_group_initializer.initializer_3d
  • colossalai.context.process_group_initializer.initializer_data
  • colossalai.context.process_group_initializer.initializer_model
  • colossalai.context.process_group_initializer.initializer_pipeline
  • colossalai.context.process_group_initializer.initializer_sequence
  • colossalai.context.process_group_initializer.initializer_tensor
  • colossalai.context.process_group_initializer.process_group_initializer
• colossalai.context.random
  • seed()
  • set_mode()
  • with_seed()
  • add_seed()
  • get_seeds()
  • get_states()
  • get_current_mode()
  • set_seed_states()
  • colossalai.context.random.seed_manager

• colossalai.context.config
  • Config
• colossalai.context.moe_context
  • MoeParallelInfo
  • MoeContext
• colossalai.context.parallel_context
  • ParallelContext
• colossalai.context.parallel_mode
  • ParallelMode
• colossalai.context.singleton_meta
  • SingletonMeta