ColossalAI
ColossalAI copied to clipboard
[autoparallel] Add alpha-beta estimation
What's new? Given a list of 1D devices, return $\alpha$ and $\beta$