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.
ColossalAI/colossalai/shardformer/policies
Frank Lee 74257cb446
[shardformer] refactored some doc and api (#4137)
1 year ago
..
__init__.py [shardformer] init shardformer code structure (#3731) 1 year ago
autopolicy.py [shardformer] import huggingface implicitly (#4101) 1 year ago
basepolicy.py [shardformer] refactored some doc and api (#4137) 1 year ago
bert.py [shardformer] refactored some doc and api (#4137) 1 year ago
bloom.py [shardformer] refactored some doc and api (#4137) 1 year ago
gpt2.py [shardformer] refactored some doc and api (#4137) 1 year ago
llama.py [shardformer] refactored some doc and api (#4137) 1 year ago
opt.py [shardformer] refactored some doc and api (#4137) 1 year ago
t5.py [shardformer] refactored some doc and api (#4137) 1 year ago
vit.py [shardformer] refactored some doc and api (#4137) 1 year ago