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
forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
Tensors and Dynamic neural networks in Python with strong GPU acceleration
License
OSU-Nowlab/pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Resources
License
Code of conduct
Security policy
Stars
Watchers
Forks
Packages 0
No packages published
Languages
- C++ 45.7%
- Python 44.1%
- Cuda 4.0%
- C 2.4%
- Objective-C++ 1.4%
- CMake 0.9%
- Other 1.5%