.. |
__init__.py
|
ed3de0c21b
Quantization + inference release (#1091)
|
3 年之前 |
async_io.py
|
e82060d090
query for libaio package using known package managers (#1250)
|
3 年之前 |
builder.py
|
b9ece257c4
update for cuda-11.4 (#1329)
|
3 年之前 |
cpu_adam.py
|
89b0fb426d
enable async io op on powerpc architectures (#1224)
|
3 年之前 |
fused_adam.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
fused_lamb.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
quantizer.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
sparse_attn.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
stochastic_transformer.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
transformer.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
transformer_inference.py
|
da1fe2f82c
Remove hard torch dependency at install (#1166)
|
3 年之前 |
utils.py
|
599258f979
ZeRO 3 Offload (#834)
|
3 年之前 |