pytorch
https://github.com/pytorch/pytorch
C++
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.
C++ not yet supported19 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- torch.nn.L1Loss works incorrectly in certain situations
- BUILD_BINARY is a lie
- Model with Caffe2 runs much slower than it with pytorch in GPU mode !!!!
- copy of Parameter converts it to Variable
- torch._utils.rebuild_cuda_tensor doesn't work with BN.num_batch_tracked
- SpectralNorm in eval doesn't connect grad to weight_orig
- Fixes for BUILD_ATEN_ONLY build flavor
- Rename potrf to cholesky
- Move collate_fn functionality / responsibility into Dataset object
- [Feature request]: add `LayerNormLSTMCell`
- Docs
- C++ not yet supported