Ammar Ahmad Awan f28432441b DeepSpeed MoE (#1310) | 3 年之前 | |
---|---|---|
.. | ||
compat.h | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 年之前 |
cpu_adam.cpp | f28432441b DeepSpeed MoE (#1310) | 3 年之前 |
custom_cuda_kernel.cu | f28432441b DeepSpeed MoE (#1310) | 3 年之前 |
fused_adam_frontend.cpp | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 年之前 |
multi_tensor_adam.cu | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 年之前 |
multi_tensor_apply.cuh | 31f46feee2 DeepSpeed JIT op + PyPI support (#496) | 4 年之前 |