You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/booster
Guangyao Zhang bdb125f83f
[doc] FP8 training and communication document (#6050)
2 months ago
..
mixed_precision
plugin [doc] FP8 training and communication document (#6050) 2 months ago
__init__.py
accelerator.py
booster.py [fp8] Merge feature/fp8_comm to main branch of Colossalai (#6016) 3 months ago