ColossalAI/colossalai/fx/passes/algorithms/__init__.py

2 lines
55 B
Python

from .ckpt_solver_chen import chen_greedy, chen_sqrtn