Matthew Hoffman
|
604d701e35
Introduce pydantic_v1 compatibility module for pydantic>=2.0.0 support (#4407)
|
1 year ago |
Olatunji Ruwase
|
4d27225f3e
zero.Init() should pin params in GPU memory as requested (#2953)
|
1 year ago |
Michael Wyatt
|
b361c72761
Update DeepSpeed copyright license to Apache 2.0 (#3111)
|
1 year ago |
Jeff Rasley
|
91d63e0228
update formatter version and style settings (#3098)
|
1 year ago |
Jeff Rasley
|
da84e60d98
add missing license info to top of all source code (#2889)
|
1 year ago |
Ammar Ahmad Awan
|
e4b3b610ba
Refactor DS inference API. No longer need replace_method. (#2831)
|
1 year ago |
Michael Wyatt
|
2e596e679d
add fix for older pydantic versions (#2611)
|
1 year ago |
Michael Wyatt
|
e59f80549e
Fix backward compatibility for InferenceConfig (#2516)
|
1 year ago |
Michael Wyatt
|
43bf035cfc
Update docs to autogenerate pydantic config model docs (#2509)
|
1 year ago |
Ammar Ahmad Awan
|
b5d18a6ab3
DeepSpeed inference config. (#2459) (#2472)
|
1 year ago |
Michael Wyatt
|
5997589683
Refactor ZeRO configs to use Pydantic (#2004)
|
2 years ago |
Karthikeyan Singaravelan
|
c7af747ce0
Import ABC from collections.abc for Python 3.10 compatibility. (#1851)
|
2 years ago |
Jeff Rasley
|
a10e4811fe
force set lf instead of crlf (https://github.com/pre-commit/pre-commit-hooks#mixed-line-ending) (#1598)
|
2 years ago |
Cheng Li
|
9caa74e577
Autotuning (#1554)
|
2 years ago |
Jeff Rasley
|
0d4a54a04d
ZeRO-Infinity (#976)
|
3 years ago |
Stas Bekman
|
c87118b0c5
[config] turn exponential notation back on for config dump (#955)
|
3 years ago |
Olatunji Ruwase
|
7bcd72a278
Make config objects json serializable (#862)
|
3 years ago |
Jeff Rasley
|
81aeea361d
Elastic training support (#602)
|
3 years ago |
Jeff Rasley
|
e5bbc2e559
Sparse attn + ops/runtime refactor + v0.3.0 (#343)
|
4 years ago |