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 supported970 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch![CodeTriage badge for pytorch/pytorch](/pytorch/pytorch/badges/users.svg?count=970)
Help out
- Issues
- [typing] Add missing `__all__` to modules.
- [aotd] Align partitioner graph output type to tuple
- Fix vulkan builds with missing overrides errors
- [inductor] Fix unsoundness with negative-valued indexing expressions
- WIP batch B,W into BW
- torch.optim: explicitly export attributes
- [dynamo]Remove stream hardcoding in dynamo VariableBuilder
- Generalization of distributed UT content to enable non cuda device execution
- Does the Distributed Data Parallel (DDP) require additional GPU memory for maintaining a model?
- torch.export.save doesn't support torch.float8_e4m3fn dtype
- Docs
- Python not yet supported