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 supported970 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch![CodeTriage badge for pytorch/pytorch](/pytorch/pytorch/badges/users.svg?count=970)
Help out
- Issues
- MSE loss triggers ZeroTensor immutable error inside `torch.func`
- Lowering after pointwise cat can lead to uncontiguous memory accesses
- gpu memory normal but one of gpus GPU-Util is 0 so cant train
- [export] fix flaky tests memory leak checks by resetting dynamo cache in test
- ppermute + collective permute
- [2.3 dynamic shapes] backend='inductor' raised: LoweringException: AssertionError: indices must be int64, byte or bool. Got [torch.float32]
- DISABLED test_pre_dispatch_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA)
- DISABLED test_retrace_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA)
- DISABLED test_serialize_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA)
- DISABLED test_aot_export_while_loop_simple_cuda_float32 (__main__.TestHOPCUDA)
- Docs
- Python not yet supported