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 supported919 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- Expose the .pt model file path during TorchScript custom class Serialization/Deserialization
- test_old_models_bc is flaky
- What changes we need to make in metrics calculation and visualization part when we use Distributed Data Parallel for distributed training
- C++ version 1.9.0 libtorch dynamic load fails -- GCC only
- Lazy modules should accept keyword arguments to forward method
- Two-element ModuleList results in error in inference_mode when jit'ed
- Replace native NHWC BN kernel
- Divergent code is needed to record usage streams on different TensorImpl types
- Sparse CSR tensor should not accept equal column indices in the same row
- Optimize torch.einsum
- Docs
- Python not yet supported