This is the NOWLAB fork of upstream PyTorch. The key features we've added are:
- Support for building PyTorch's distributed module with any [CUDA/ROCm]-aware MPI library including NOWLAB's MVAPICH2-GDR
- Support for MPI + fp16 communication operations
OSU-Nowlab/pytorch
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
Repository files navigation
Languages
- C++ 45.7%
- Python 44.1%
- Cuda 4.0%
- C 2.4%
- Objective-C++ 1.4%
- CMake 0.9%
- Other 1.5%