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 supported925 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- JIT is not compatible with data parallel
- [Feature Request] cdist: pairwise distances between two sets of tensors with batch mode
- manylinux2014 compatible wheels
- caffe2: multi-thread(or multi-instance) predict is much slower than the single thread or single instance
- Gather backward is faster than integer indexing on GPU
- [JIT] jit.trace fails with custom GRUs and CUDA when the sequence is longer
- [C++/Pytorch] Get input shapes
- get/set device in c++
- [Feature request] create sparse coo matrix w/o index check
- [discussion] Recommend a different file extension for models (.PTH is a special extension for Python)
- Docs
- Python not yet supported