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 supported1144 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [CUDA] MultiheadAttention with masks and dropout produces NaNs
- [FX] Add distributed support for codegen dump: rank suffix + atomic write
- [spmd_types] pipeline parallel support for spmd_types-annotated tensors
- Python 3.15 support for PyTorch
- [dynamo, nested graph breaks] add strong nested graph break test infrastructure
- fix(core): support ldexp on custom backends by falling back to decomposition
- optim: validate ns_steps and eps in Muon.__init__
- `torch.compile(mode='reduce-overhead')` fails when user model uses wait_stream to synchronize side work
- [inductor] Fix missing libtorch_python symbols in FBCODE when using cpp-wrapper (#185551)
- [ROCm] Enable linalg tests for eig, ldl_solve operator & enable test_linalg_solve, test_triangular_solve linalg tests with cuBLAS path
- Docs
- Python not yet supported