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 supported922 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- PyTorch not recognizing GPU on WSL - installed cudnn and cuda
- upstream `apex.normalization.FusedRMSNorm`
- torch.nn.LayerNorm support for arbitrary axis in order to allow NCHW application
- Fancy indexing bug when combining masks with indexes
- NCCL Backend does not support ComplexFloat data type
- torch.linalg.lstsq is nondeterministic
- Remove deprecated torch.chain_matmul
- [feature request] `torch.to(obj, device)` supporting recursive lists/dicts/tuples of tensors probably by uplifting/promoting `torch.distributed.utils._recursive_to`
- Lazy Tensor Core Documentation Out-of-Date
- [feature request] quantized and low-level int8 operators (matmul, gemm etc) on CUDA + integrate LLM.int8 + integrate ZeroQuant?
- Docs
- Python not yet supported