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/amp/naive_amp
flybird11111 af6aa9ed06
[plugin] hybrid support zero bubble pipeline (#6060)
2 months ago
..
grad_scaler [npu] change device to accelerator api (#5239) 11 months ago
mixed_precision_mixin [plugin] hybrid support zero bubble pipeline (#6060) 2 months ago
__init__.py [legacy] clean up legacy code (#4743) 1 year ago
mixed_precision_optimizer.py [plugin] hybrid support zero bubble pipeline (#6060) 2 months ago