colossalai/autochunk - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	autochunk_codegen.py	25.1 KB
	estimate_memory.py	9.1 KB
	reorder_graph.py	4.6 KB
	search_chunk.py	11.4 KB
	select_chunk.py	7.9 KB
	trace_flow.py	20.3 KB
	trace_indice.py	34.6 KB
	utils.py	7.0 KB