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
..
csrc	[zero] cpu adam kernel (#288)	2022-03-11 15:50:28 +08:00
__init__.py	refactor kernel (#142)	2022-01-13 16:47:17 +08:00
layer_norm.py	moved env variables to global variables; (#215)	2022-02-15 11:31:13 +08:00
multihead_attention.py	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00
scaled_softmax.py	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00