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 supported1202 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- nn.functional.threshold OpInfo reference raises OverflowError on unsigned dtypes with NumPy >= 2.5
- Inductor doesn't do a good job of fusing Domino CE
- [inductor] Eliminate singleton one-hot reductions
- [Inductor]resize_() causes CUDA Graph memory pool error with mode="max-autotune"
- [RFC] Transition ROCm nightly wheels to wheel-based ROCm install (7.14+)
- [profiler] profile_all_threads misses memory events from worker threads
- [Testcase Refactoring] Migrate test_fsdp_fine_tune to capability gating
- [Testcase Refactoring] Migrate test_fsdp_flatten_params to capability gating
- [Testcase Refactoring] Demote test_fsdp_fx to Strategy 1 / GENERIC
- [Testcase Refactoring] Migrate test_fsdp_exec_order to capability gating
- Docs
- Python not yet supported