Author | SHA1 Message | Date |
---|---|---|
Jeff Rasley | da1fe2f82c Remove hard torch dependency at install (#1166) | 3 years ago |
Reza Yazdani | 26e3841cd4 Change the sparse attention API to be compatible with latest changes of triton (#902) | 3 years ago |
Samyam Rajbhandari | 599258f979 ZeRO 3 Offload (#834) | 3 years ago |
Jeff Rasley | 24e07399fc update SA comp check to fix torch-cpu issue (#631) | 3 years ago |
Jeff Rasley | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 years ago |