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 supported918 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- Complex Number support for distributed
- with torch.cuda.amp.autocast() get out of memory error when using with torch.no_grad() during validation
- SummaryWriter deletes data automatically when there are too many
- Authentication for RPC
- How to use clang as a cuda compiler instead of nvcc?
- Support deleting a parameter/buffer by name
- Autograd support for the tensor multiplication of sparse tensors
- Trying to compile PyTorch for sm_30 fails with `error: identifier “__ldg” is undefined`
- Quantization - we need a better solution for tracking quantization backend settings in a model
- Mixing Numpy's arrays and PyTorch tensors
- Docs
- Python not yet supported