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 supported1027 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [inductor][cpu] Models performance regression in 2024-08-04 nightly release
- return state dict without optimized module
- Support for ROCm 6.2
- Add support for other backends in get_preferred_device
- [xla hash update] update the pinned xla hash
- [inductor] Add type hints to functions in mkldnn_fusion.py
- [DEBUG] Strange behavior observed with PyTorch 2.4.0 + Windows + CPU inference
- Add compiler bisector
- Add support for more dtypes for serialization
- Change signature of CompilerFn for register_backend decorator
- Docs
- Python not yet supported