Jeff Rasley
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 year ago |
Ma, Guokai
|
98cc35b6a8
Abstract accelerator (step 3) (#2677)
|
1 year ago |
Jeff Rasley
|
35575bce4e
remove print side effect from importing deepspeed (#2704)
|
1 year ago |
cassieesvelt
|
a3d7f106f5
Add mlflow logging for aml (#2495)
|
1 year ago |
Ikko Ashimine
|
6273dffc2f
Fix typo in autotuner.py (#2639)
|
1 year ago |
Rahil Bathwal
|
384f17b05b
[deepspeed/autotuner] Bug fix for binary search for batch size (#2162)
|
1 year ago |
Cheng Li
|
abe4fc6b55
encoded ds config into command line argument when launching child processes in autotuning (#2524)
|
1 year ago |
Cheng Li
|
8da0238b7a
rollback ds config changes (#2395)
|
2 years ago |
Rahil Bathwal
|
ee5ce52460
fix missing import (#2175)
|
2 years ago |
Michael Wyatt
|
5997589683
Refactor ZeRO configs to use Pydantic (#2004)
|
2 years ago |
Alex Hedges
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 years ago |
Cheng Li
|
cccc5450e0
Fix autotuning config (#1660)
|
2 years ago |
Alex Hedges
|
4cf970e6bb
Add codespell to pre-commit checks (#1717)
|
2 years ago |
Jeff Rasley
|
938449e34a
[autotuning] guard tabulate package import (#1569)
|
2 years ago |
Cheng Li
|
9caa74e577
Autotuning (#1554)
|
2 years ago |