mirror of https://github.com/hpcaitech/ColossalAI
e7d3afc9cc
* [optimizer] add div_scale for optimizers * [zero] use div_scale in zero optimizer * fix testing error |
||
---|---|---|
.. | ||
cuda_native | ||
jit | ||
__init__.py |