Reza Yazdani 8164ea9e6d Fixing several bugs in the inference-api and the kernels (#1951) 2 年之前
..
__init__.py c3c8d5dd93 AMD support (#1430) 2 年之前
async_io.py 91e15593ea Control ds_report output (#1622) 2 年之前
builder.py 73b5d9833a Fix setup.py crash when torch is not installed. (#1866) 2 年之前
cpu_adagrad.py c3c8d5dd93 AMD support (#1430) 2 年之前
cpu_adam.py 737fee63ba Update cpu_adam.py (#1968) 2 年之前
fused_adam.py 73b5d9833a Fix setup.py crash when torch is not installed. (#1866) 2 年之前
fused_lamb.py 73b5d9833a Fix setup.py crash when torch is not installed. (#1866) 2 年之前
quantizer.py da1fe2f82c Remove hard torch dependency at install (#1166) 3 年之前
sparse_attn.py c3c8d5dd93 AMD support (#1430) 2 年之前
stochastic_transformer.py da1fe2f82c Remove hard torch dependency at install (#1166) 3 年之前
transformer.py b4fcd98ff0 Inference PP changes for neox (#1899) 2 年之前
transformer_inference.py 8164ea9e6d Fixing several bugs in the inference-api and the kernels (#1951) 2 年之前
utils.py 599258f979 ZeRO 3 Offload (#834) 3 年之前