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 supported1101 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- Adjust test_linalg_lstsq_input_check to be compatible with other devices
- [inductor] Skip addcdiv decomposition for AdamW bitwise precision
- rewrite all remaining size_hint usages in ir.py to support unbacked and use new APIs
- [ROCm] allow PYTORCH_AOTRITON_COMMIT env var to override git hash
- MobileBertForMaskedLM is 90% slower with unbacked vs backed !
- Fix pandas/numpy binary incompatibility breaking test_dataloader and test_datapipe
- [dynamo] Dynamo distributed collective rewrite should not change eager return semantics
- [dynamic shapes] unbacked pool unpool
- [0/N] Update `@torch.compile` calls to use `backend="eager"` in dynamo tests
- Fix clang build on aarch64
- Docs
- Python not yet supported