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 supported1001 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- Calling torch.linalg.cholesky on a CPU tensor requires compiling PyTorch with LAPACK.
- Complex-Valued Gaussian distributions
- Test public bindings in CI gives weird output on error
- torch.Tag doesn't have accurate mypy info
- Torch1.10.2 is slower than torch1.9.1
- Emulating FP64 and increased precisions on Apple silicon
- bmm operator in bfloat16 has low TFLOPS for some tensor shapes with CUDA 11.6
- RPC: wait method of Future object return 0 sometimes in rpc framework
- torch.nn.Upsample's error message is inconsistent with the documentation
- During DDP training timm densenet121, mobilenetv2(v3) models do not save state_dict correctly.
- Docs
- Python not yet supported