mirror of https://github.com/hpcaitech/ColossalAI
933b6c6367
* [fx] add pofo algorithm * [fx] Add pofo solver * [fx] code refactor * [fx] fix test_linearize import |
||
---|---|---|
.. | ||
__init__.py | ||
ckpt_solver_chen.py | ||
ckpt_solver_pofo.py | ||
ckpt_solver_rotor.py | ||
linearize.py | ||
operation.py |