Colossal-AI API documentation
API REFERENCE
- colossalai
- colossalai.amp
- colossalai.builder
- colossalai.cli
- colossalai.communication
- colossalai.context
- colossalai.engine
- colossalai.fx
- colossalai.gemini
- colossalai.kernel
- colossalai.logging
- colossalai.nn
- colossalai.pipeline
- colossalai.registry
- colossalai.tensor
- colossalai.testing
- colossalai.trainer
- colossalai.utils
- colossalai.zero
- colossalai.constants
- colossalai.core
- colossalai.global_variables
- colossalai.initialize
Useful links for Colossal-AI