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
- [preproc] shared lock for jit
- [autograd] Implement backward and forward AD for torch.aminmax
- Annotate runtime exceptions from AOTAutograd with CompileId
- Fix PyTorchStreamWriter crash on write failure (e.g. disk full)
- Block mask caching for flex attention and SAC don't play nicely together (RuntimeError: Only Tensors of floating point and complex dtype can require gradients)
- Improve FX partitioner complexity
- [ROCm][Distributed] Replace MI300-only ROCm gating with platform capability checks
- [HOP][print]Add grad printing with flag
- fixed issue #146018
- [dynamo] Add polyfills for operator.concat and operator.iconcat
- Docs
- Python not yet supported