vtlam

Pinned Repositories

torch_ucc
Pytorch process group third-party plugin for UCC
Language:C++20 66 622
nccl
Optimized primitives for collective multi-GPU communication
Language:C++3.4k 153 1.4k846
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:Python85.7k 1.8k 48k23.1k
nccl
Optimized primitives for collective multi-GPU communication
Language:C++00
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++00
torch_ucc
Pytorch process group third-party plugin for UCC
Language:C++00

vtlam's Repositories

vtlam/nccl
Optimized primitives for collective multi-GPU communication
Language:C++00
vtlam/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Language:C++00
vtlam/torch_ucc
Pytorch process group third-party plugin for UCC
Language:C++00