colossalai/fx/passes - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	experimental
	__init__.py	266 B
	adding_split_node_pass.py	13.2 KB
	concrete_info_prop.py	12.0 KB
	meta_info_prop.py	13.7 KB
	passes_for_gpt2_test.py	15.5 KB
	shard_1d_pass.py	6.7 KB
	split_module.py	13.4 KB
	utils.py	6.0 KB