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 supported1107 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [Inconsitency] Numerical Discrepancy Between Eager and Inductor Backends for atan + special_psi Composition
- Fix off-by-one error in upsample_nearest decomposition
- fix meta for fft_c2c
- [dynamo] Support leaf_function on nn.Module methods
- [inductor] Improve 1D pointwise kernel heuristic on B200
- [inductor] Decompose mm to pointwise mul when K==1
- `torch._scaled_mm`: better default output dtype for MXFP8
- [DTensor] Add global fix for 0D tensor (scalar) support in sharding pā¦
- Don't redefine `TORCH_USE_CUDA_DSA`
- [docker] Speed up CI docker image builds
- Docs
- Python not yet supported