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 supported1216 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [CI] Re-enable disabled CI jobs when PR fixes the disable issue
- [trymerge] Don't wait on a queued Import Status once CodeSync clears the commit
- Fix Dynamo silently dropping custom jvp of autograd.Function under forward AD
- [inductor][cudagraphs] Prevent SIGSEGV for mutated outputs reused across generations
- [ROCm] Add Context::tf32Supported() guard for hipBLASLt GEMMs
- Fix _weight_norm_interface for g, v, and layout cases it mishandled
- Fix backward for reductions with a complex dtype= on a real input
- [Testcase Refactoring] 1. Add instantiate_device_type_tests to generalize device types. 2. Add proper hw_classification attribute to test classes.
- [Test] Make test_embedding device-agnostic for OOT backends
- Set-family VariableTracker hierarchy diverges from CPython
- Docs
- Python not yet supported