4 Commits (dd7cc5829998c26f3207bee578f2d8470e7f61f2)

Author	SHA1	Message	Date
Frank Lee	70c58cfd4f	[shardformer] supported fused qkv checkpoint (#4073)	2023-07-04 16:05:01 +08:00
Frank Lee	8eb09a4c69	[shardformer] support module saving and loading (#4062) ... * [shardformer] support module saving and loading * polish code	2023-07-04 16:05:01 +08:00
Frank Lee	45d9384346	[shardformer] removed inplace tensor sharding (#4018)	2023-07-04 16:05:01 +08:00
Frank Lee	015af592f8	[shardformer] integrated linear 1D with dtensor (#3996) ... * [shardformer] integrated linear 1D with dtensor * polish code	2023-07-04 16:05:01 +08:00