Colossal-AI API documentation

API REFERENCE

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

Useful links for Colossal-AI

• Colossal-AI Examples
• Colossal-AI Benchmarks
• Colossal-AI Tutorial

Indices and tables

• Index