cpu-inference.yml
- Files15
- .github/workflows
- formatting.yml
- nv-accelerate-v100.yml
- nv-inference.yml
- nv-lightning-v100.yml
- nv-megatron.yml
- nv-mii.yml
- nv-pre-compile-ops.yml
- nv-torch-latest-cpu.yml
- nv-torch-latest-v100.yml
- nv-transformers-v100.yml
- python.yml
- csrc
- lamb
fused_lamb_cuda.cpp
- transformer
- ds_transformer_cuda.cpp
- deepspeed/comm
comm.py