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 supported1207 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [Testcase Refactoring] Add proper hw_classification attribute to test classes
- [MPS] Use int32 scan in flat unique when counts fit (faster on Apple GPU)
- [CUDAGraph] Reuse blocks across parent and conditional body captures
- [Testcase Refactoring] Add hw_classification and migrate test_c10d_ucc.py to MultiProcContinuousTest
- [xpu][skip] update skip all platform to skip specific platform for test_max_reads_limits_fusion
- [Inductor] Fallback complex tensor constants on Triton
- [Testcase Refactoring] Use is_backend_available for MultiProcContinuousTest backend check
- fix: Silent Out-of-Bounds Read for Strided/OffSet Compiled Kernels
- [Testcase Refactoring] Refactor test_stage with instantiate_device_type_tests and hw-classification
- [MPS] Honor use_fast_math in Inductor Metal codegen
- Docs
- Python not yet supported