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 supported1106 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- General MPS op coverage tracking issue
- AVX512 and Vec512
- GPU acceleration for Apple's M1 chip?
- nn.MultiheadAttention causes gradients to become NaN under some use cases
- Mixed precision causes NaN loss
- Version 1.3 no longer supporting Tesla K40m?
- Python/C++ API Parity: torch.nn modules and functional
- Download speed issues with the pytorch conda channel
- DataLoader num_workers > 0 causes CPU memory from parent process to be replicated in all worker processes
- [feature request] Add matrix functions
- Docs
- Python not yet supported