colossalai/tensor - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	d_tensor
	moe_tensor
	padded_tensor
	__init__.py	600 B
	colo_parameter.py	3.5 KB
	colo_tensor.py	3.3 KB
	comm_spec.py	21.1 KB
	param_op_hook.py	5.3 KB
	shape_consistency.py	35.0 KB
	sharding_spec.py	11.7 KB
	utils.py	8.3 KB