Alex Hedges
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 years ago |
Ammar Ahmad Awan
|
36ad3119d5
DeepSpeed comm backend v1 (#1985)
|
2 years ago |
Justin Chiu
|
4912e0ad7e
Various ZeRO Stage3 Optimizations + Improvements (including bfloat16 support) (#1453)
|
2 years ago |
Alex Hedges
|
fc2f378ece
Improve pre-commit hooks (#1602)
|
2 years ago |
Jeff Rasley
|
8db4fdf815
disable pipe test (#915)
|
3 years ago |
Shaden Smith
|
e59ba12d4d
make test_pipe more stable (#683)
|
3 years ago |
Olatunji Ruwase
|
da5563a9c1
LR scheduler unit tests (#429)
|
3 years ago |
Jeff Rasley
|
eec44af1e3
Turn back on PP tests (#558)
|
3 years ago |
Jeff Rasley
|
679fc13512
turning off different tests (temp)
|
4 years ago |
Shaden Smith
|
65c2f974d8
Pipeline parallel training engine. (#392)
|
4 years ago |