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 supported1189 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [CUDA] Warn instead of assert in `reportProcessMemoryInfo` to prevent crash on Tegra OOM
- [DTensor][local_map]Support `None` placements in `local_map` placement specs
- Output Shape Check added to transposed Conv
- Support input mutations in CUDA graph control flow.
- torch.func.jacfwd gives incorrect Jacobian for torch.pow(torch.abs(complex), complex exponent)
- torch.linalg.qr (float64) repeated in a loop crashes the process (SIGSEGV, exit 139) on ROCm 7.x
- Lower symbolic bool not to torch.sym_not
- Add test cases for 0-sized tensor padding
- Support compiled async all_reduce wait
- Clear Dynamo weakrefs before graph-break resumes
- Docs
- Python not yet supported