pytorch
https://github.com/pytorch/pytorch
Python
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.
Python not yet supported970 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch![CodeTriage badge for pytorch/pytorch](/pytorch/pytorch/badges/users.svg?count=970)
Help out
- Issues
- MPS slows down after sleep
- [RFC] Support Torchrun for Out-of-the-tree Devices
- CUDA kernel failed: no kernel image is available for execution on the device
- torch.compile error
- [inductor] refine loop split logic
- fix Invalid call to aoti_torch_tensor_copy_ #123039
- Typinghints on tensor arithmetic operations (other:Any) cause incompatibilities
- Expand stale management to Issues
- Fix OOMs during parallel builds
- [FSDP+TP] RuntimeError: 'weight' must be 2-D
- Docs
- Python not yet supported