зеркало из https://github.com/microsoft/DeepSpeed.git
ccfdb84e2a
The user interface: https://github.com/microsoft/DeepSpeed-MII/pull/433 nv-a6000 ci running against the MII branch linked above is [here](https://github.com/microsoft/DeepSpeed/actions/runs/8192124606) Co-authored-by: Zhen Zheng [zhengzhen@microsoft.com](mailto:zhengzhen@microsoft.com) Co-authored-by: Shiyang Chen [csycfl@gmail.com](mailto:csycfl@gmail.com) Co-authored-by: Arash Bakhtiari [abakhtiari@microsoft.com](mailto:abakhtiari@microsoft.com) Co-authored-by: Haojun Xia [xhjustc@mail.ustc.edu.cn](mailto:xhjustc@mail.ustc.edu.cn) --------- Co-authored-by: ZHENG, Zhen <zhengzhen.z@qq.com> Co-authored-by: Shiyang Chen <csycfl@gmail.com> Co-authored-by: Haojun Xia <xhjustc@mail.ustc.edu.cn> Co-authored-by: Arash Bakhtiari <arash@bakhtiari.org> Co-authored-by: Michael Wyatt <mrwyattii@gmail.com> Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com> |
||
|---|---|---|
| .. | ||
| cpu | ||
| hpu | ||
| npu | ||
| xpu | ||
| __init__.py | ||
| all_ops.py | ||
| async_io.py | ||
| builder.py | ||
| cpu_adagrad.py | ||
| cpu_adam.py | ||
| cpu_lion.py | ||
| evoformer_attn.py | ||
| fused_adam.py | ||
| fused_lamb.py | ||
| fused_lion.py | ||
| inference_core_ops.py | ||
| inference_cutlass_builder.py | ||
| quantizer.py | ||
| ragged_ops.py | ||
| ragged_utils.py | ||
| random_ltd.py | ||
| sparse_attn.py | ||
| spatial_inference.py | ||
| stochastic_transformer.py | ||
| transformer.py | ||
| transformer_inference.py | ||