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 7f8b16635b
[misc] refactor launch API and tensor constructor (#5666)
7 months ago
..
d_tensor [misc] refactor launch API and tensor constructor (#5666) 7 months ago
moe_tensor [moe] support mixtral (#5309) 10 months ago
padded_tensor [shardformer] refactor embedding resize (#5603) 7 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