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 supported915 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- [FYI] MultiheadAttention / Transformer
- Ability to download docs HTML for offline use
- Problem with multiprocessing, custom __getstate__ with Tensors and forkserver
- Document different gradient wrt to jax when nesting
- Interrupting a DDP worker while using the CUDA MPS server causes CUDA to hang until reboot
- [feature request] Out-variant and dtype argument for torch.argmax / torch.argmin / torch.argsort (and friends)
- torch.nn.functional.normalize epsilon too small for half precision
- TensorRT: CheckDims() need adjustment for EXPLICIT_BATCH?
- quantization - Missing operations needed for object detection
- addition of attention based techniques to pytorch
- Docs
- Python not yet supported