You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

LuGY 6a3f9fda83 [cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497)		3 years ago
..
cuda_native	[cuda] modify the fused adam, support hybrid of fp16 and fp32 (#497)	3 years ago
jit	adapted for sequence parallel (#163)	3 years ago
__init__.py	refactor kernel (#142)	3 years ago