ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

History

Jiarui Fang e761ad2cd7 Revert "[zero] add ZeroTensorShardStrategy (#793 )" (#806 )		2022-04-19 14:40:02 +08:00
..
csrc	Revert "[zero] add ZeroTensorShardStrategy (#793 )" (#806 )	2022-04-19 14:40:02 +08:00
__init__.py	…
layer_norm.py	…
multihead_attention.py	…
scaled_softmax.py	…