colossalai/tensor/d_tensor - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	__init__.py	1.2 KB
	api.py	18.2 KB
	comm_spec.py	10.9 KB
	layout_converter.py	26.9 KB
	layout.py	2.7 KB
	misc.py	231 B
	README.md	3.8 KB
	sharding_spec.py	9.6 KB
	utils.py	3.2 KB