ColossalAI/colossalai/kernel
HELSON e7d3afc9cc
[optimizer] add div_scale for optimizers (#2117)
* [optimizer] add div_scale for optimizers

* [zero] use div_scale in zero optimizer

* fix testing error
2022-12-12 17:58:57 +08:00
..
cuda_native [optimizer] add div_scale for optimizers (#2117) 2022-12-12 17:58:57 +08:00
jit [kernel] added jit warmup (#1792) 2022-11-08 16:22:23 +08:00
__init__.py