tests/test_moe - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

41374 0 0 Python

	moe_utils.py	3.1 KB
	test_deepseek_layer.py	2.3 KB
	test_kernel.py	3.2 KB
	test_mixtral_layer.py	2.2 KB
	test_moe_checkpoint.py	6.0 KB
	test_moe_ep_tp.py	4.4 KB
	test_moe_ep_zero.py	3.8 KB