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 supported1107 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [Inductor] Missing host-side synchronization after non-blocking D2H copies
- [PyTorch]: Fix pinned host allocator active counter updates
- Support torch.accelerator.get_device_capability on CUDA
- Use std:: random distributions wherever possible
- Add error message when autocast is misused as a function wrapper
- [Pallas] Fix argmax/argmin with keepdim on 3D+ tensors
- [cudagraph_trees] Fix tensor_weakrefs/stack_traces assertion in dealloc_current_path_weakrefs
- [SymmMem] Add thread safety to NCCL and NVSHMEM backends
- Fix SVE detection on older versions of GCC
- Add mps support for addmm relu
- Docs
- Python not yet supported