Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
COMMITS
/ tests/plugins/test_amp_plugins.py June 16, 2021
Y
Make optimizers skippable when using amp (#7975)
Yifu Wang committed
April 30, 2021
T
[bugfix] Apex never instantiated. (#7274)
thomas chaton committed
April 13, 2021
A
Clean up environment access in plugins (#6941)
Adrian Wälchli committed
March 3, 2021
J
Simplify test for AMP plugins (#6311)
Jirka Borovec committed