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/tensor
Hongxin Liu 641b1ee71a
[devops] remove post commit ci (#5566)
8 months ago
..
d_tensor [devops] remove post commit ci (#5566) 8 months ago
moe_tensor [moe] support mixtral (#5309) 10 months ago
__init__.py
colo_parameter.py [gemini] fix param op hook when output is tuple (#5355) 10 months ago
colo_tensor.py
comm_spec.py
param_op_hook.py [gemini] fix param op hook when output is tuple (#5355) 10 months ago
shape_consistency.py
sharding_spec.py
utils.py