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 supported114 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- Remove THCHalfAutoNumerics.cuh
- Improve binary release for PyTorch domain library
- When use pin_memory, dataloader can get stuck inside pin_memory
- Successive Layer Normalization in nn.Transformer
- max_pool2d cuda should have channel last optimized kernels[Performance improvement]
- CUDA_KERNEL_LOOP: prevent int overflow in loop increment.
- Implement `cpu_serial_kernel` and `cpu_serial_kernel_vec` for TensorIterator
- Implement operators logical_and and logical_or
- Enable log_softmax and CrossEntropyLoss for bfloat16
- Illegal memory access occurs when using nn.AvgPool2d
- Docs
- C++ not yet supported