Colossal-AI API documentation

API REFERENCE

• colossalai
  • colossalai.amp
  • colossalai.builder
  • colossalai.communication
  • colossalai.context
  • colossalai.engine
  • colossalai.kernel
  • colossalai.logging
  • colossalai.nn
  • colossalai.registry
  • colossalai.trainer
  • colossalai.utils
  • colossalai.zero
  • colossalai.initialize

Indices and tables

• Index