ColossalAI/colossalai/kernel
LuGY a3269de5c9 [zero] cpu adam kernel (#288)
* Added CPU Adam

* finished the cpu adam

* updated the license

* delete useless parameters, removed resnet

* modified the method off cpu adam unittest

* deleted some useless codes

* removed useless codes

Co-authored-by: ver217 <lhx0217@gmail.com>
Co-authored-by: Frank Lee <somerlee.9@gmail.com>
Co-authored-by: jiaruifang <fangjiarui123@gmail.com>
2022-03-11 15:50:28 +08:00
..
cuda_native [zero] cpu adam kernel (#288) 2022-03-11 15:50:28 +08:00
jit adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
__init__.py refactor kernel (#142) 2022-01-13 16:47:17 +08:00