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

4 lines
122 B
Python

from .ckpt_solver_chen import chen_greedy
from .linearize import linearize
from .ckpt_solver_rotor import solver_rotor