Jeff Rasley
|
2665c8b149
Fix 1bit extra issue (#1542)
|
2 years ago |
Alex Hedges
|
be789b1665
Fix many typos (#1423)
|
3 years ago |
Anurag Kumar
|
8e577c923d
Update setup.py (#1361)
|
3 years ago |
Adam Moody
|
e82060d090
query for libaio package using known package managers (#1250)
|
3 years ago |
Jeff Rasley
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 years ago |
eltonzheng
|
10104284a2
Create symlinks on Windows setup (#1099)
|
3 years ago |
Reza Yazdani
|
ed3de0c21b
Quantization + inference release (#1091)
|
3 years ago |
Samyam Rajbhandari
|
599258f979
ZeRO 3 Offload (#834)
|
3 years ago |
Jeff Rasley
|
81aeea361d
Elastic training support (#602)
|
3 years ago |
Jeff Rasley
|
be33bea475
Add compute capability 8.0 if on cuda 11+ (#572)
|
3 years ago |
Shaden Smith
|
6009713653
Adds long_description to setup.py (#560)
|
3 years ago |
Seunghwan Hong
|
d81cb26d92
Fix setup.py for cpu-only environment installation (#538)
|
3 years ago |
Jeff Rasley
|
ca9ab1201f
ds_report bug fix on cpu and guard torch import in setup.py (#524)
|
4 years ago |
Jeff Rasley
|
31f46feee2
DeepSpeed JIT op + PyPI support (#496)
|
4 years ago |
Jeff Rasley
|
7ddfda8526
Add support for p100 in transformer kernels (#470)
|
4 years ago |
Jeff Rasley
|
1afca8f722
revert previous (accidental) change
|
4 years ago |
Jeff Rasley
|
b8eb40eb67
add compute cap of 6.0 to transformer kernels
|
4 years ago |
Olatunji Ruwase
|
7b8be2a7d9
Disable default installation of CPU Adam (#450)
|
4 years ago |
Shaden Smith
|
5812e84544
readthedocs yaml configuration (#410)
|
4 years ago |
Jeff Rasley
|
240ea97b33
only add 1bit adam reqs if mpi is installed, update cond build for cpu-adam (#400)
|
4 years ago |
Jeff Rasley
|
41db1c2f03
ZeRO-Offload release (#391)
|
4 years ago |
Ammar Ahmad Awan
|
01726ce2b8
Add 1-bit Adam support to DeepSpeed (#380)
|
4 years ago |
Jeff Rasley
|
e5bbc2e559
Sparse attn + ops/runtime refactor + v0.3.0 (#343)
|
4 years ago |
Jeff Rasley
|
871f7e6305
Update setup.py (#298)
|
4 years ago |
Jeff Rasley
|
734d8991c8
Transformer kernel release (#242)
|
4 years ago |
Jeff Rasley
|
3d3f8d36a4
PyTorch 1.3+ build support (#135)
|
4 years ago |
Shaden Smith
|
50ae149f82
Moving to major/minor/patch versioning. (#51)
|
4 years ago |
Jeff Rasley
|
20ff66a0c1
Azure tutorial updates and cleanup (#43)
|
4 years ago |
Jeff Rasley
|
a5acb5b22d
updating formatting to pass yapf
|
4 years ago |
Jeff Rasley
|
26e5ecab9b
Initial commit of setup
|
4 years ago |