colossalai/zero/gemini - hpcaitech/ColossalAI - Morph

SIGN IN SIGN UP

hpcaitech / ColossalAI UNCLAIMED

Making large AI models cheaper, faster and more accessible

0 0 96 Python

	chunk
	memory_tracer
	__init__.py	490 B
	gemini_ddp.py	44.5 KB
	gemini_hook.py	4.2 KB
	gemini_mgr.py	7.5 KB
	gemini_optimizer.py	38.0 KB
	placement_policy.py	11.7 KB
	utils.py	4.1 KB