ColossalAI/colossalai/kernel
Jiarui Fang e761ad2cd7
Revert "[zero] add ZeroTensorShardStrategy (#793)" (#806)
2022-04-19 14:40:02 +08:00
..
cuda_native Revert "[zero] add ZeroTensorShardStrategy (#793)" (#806) 2022-04-19 14:40:02 +08:00
jit adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
__init__.py fix format (#568) 2022-04-06 11:40:59 +08:00