pytorch
https://github.com/pytorch/pytorch
C++
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
C++ not yet supported651 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- MPS crashes in `.abs()` for complex dtype (FFT outputs)
- [FSDP] .modules() return original modules instead of FSDP prefixed modules
- Segmentation fault after permute on MPS
- toleranceOverride should override atol and rtol even when explicitly specified in a test
- [primTorch] prims are being checked as part of the BC check
- Can't convert nn.multiheadAttetion(q,k,v) to Onnx when key isn't equal to value
- Python memory allocator called without holding the GIL when running torchrun under Python debug version
- DDP multi host with single GPU each.
- Static initializer update
- [primTorch] many primTorch test xfails are due to chalf
- Docs
- C++ not yet supported