Logan Adams
|
8fb111c08d
Treat empty environment variables as unset in (#4185)
|
1 year ago |
Yejing-Lai
|
7290aace9b
[CPU] Skip CPU support unimplemented error (#3633)
|
1 year ago |
stephen youn
|
69d1b9f978
DeepSpeed-Triton for Inference (#3748)
|
1 year ago |
Tian, Feng
|
6938c449de
Add snip_momentum structured pruning which can support higher sparse ratio with minor accuracy loss (#3300)
|
1 year ago |
Pablo Emídio S.S
|
f3f4c44959
Build: Update license in setup (#3484)
|
1 year ago |
Michael Wyatt
|
bcccee4d85
Fix cupy install version detection (#3276)
|
1 year ago |
Gavin Goodship
|
adc15e1c17
Update curriculum-learning.md (#3031)
|
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
|
bbfd0a6a3e
update email info
|
1 year ago |
Logan Adams
|
b4d40e357b
Fix example command when building wheel with dev version specified (#2815)
|
1 year ago |
Jeff Rasley
|
0b549ad70a
[install] only add deepspeed pkg at install (#2714)
|
1 year ago |
Jeff Rasley
|
cd271a4aa6
exclude benchmarks during install (#2698)
|
1 year ago |
Ma, Guokai
|
9548d48f48
Abstract accelerator (step 2) (#2560)
|
1 year ago |
Jeff Rasley
|
35eabb0a33
Fix issues w. python 3.6 + add py-version checks to CI (#2589)
|
1 year ago |
Michael Wyatt
|
521d329b97
Fix CI issues related to cupy install (#2483)
|
1 year ago |
eltonzheng
|
b85eb3b979
Fix build issues on Windows (#2428)
|
2 years ago |
Jeff Rasley
|
1b7c6791d5
only add deps if extra is explictly called (#2432)
|
2 years ago |
Alex Hedges
|
316c4a43e0
Add flake8 to pre-commit checks (#2051)
|
2 years ago |
Alex Hedges
|
3540ce74d9
Check for bf16 support only if CUDA is available (#2049)
|
2 years ago |
Quentin Anthony
|
9b70ce56e7
Comms Benchmarks (#2040)
|
2 years ago |
Jeff Rasley
|
7c3344e215
DeepSpeed examples refresh (#2021)
|
2 years ago |
Jeff Rasley
|
b666d5cd73
[inference] test suite for ds-kernels (bert, roberta, gpt2, gpt-neo, gpt-j) (#1992)
|
2 years ago |
Michael Wyatt
|
7fc3065074
Add torch-latest and torch-nightly CI workflows (#1990)
|
2 years ago |
Jithun Nair
|
350d74ca39
Invoke hipify from op builder for JIT extension builds (#1807)
|
2 years ago |
Andrii Oriekhov
|
d7684f4e81
add GitHub URL for PyPi (#1812)
|
2 years ago |
Jeff Rasley
|
c3c8d5dd93
AMD support (#1430)
|
2 years ago |
Jeff Rasley
|
9351266f78
Multi-node save pid support + allow sparse-attn extra (#1728)
|
2 years ago |
Jeff Rasley
|
e46d808a1b
MoE inference + PR-MoE model support (#1705)
|
2 years ago |
Cheng Li
|
9caa74e577
Autotuning (#1554)
|
2 years ago |