Colossal-AI
feature/docs
API REFERENCE
Colossal-AI
colossalai.amp
colossalai.amp.apex_amp
colossalai.amp.naive_amp
colossalai.amp.torch_amp
colossalai.amp.amp_type
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
Colossal-AI
»
Colossal-AI
»
colossalai.amp
»
colossalai.amp.amp_type
Edit on GitHub
colossalai.amp.amp_type
class
colossalai.amp.amp_type.
AMP_TYPE
(
value
)
[source]
An enumeration.