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 supported906 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- [GradScaler] Inconsistent scale values across different GPUs caused by uneven inputs for AMP DDP training
- Can not use x=torch.tensor(b), to create a Tensor out of a List[List[Tensor]] (A List of Lists of Tensors)
- Calling item() on symbolic shape fake tensor should give more clear error message
- Wrong output type hint for `F.one_hot`
- No pytorch_jni.dll file in libtorch 1.13.0 lib folder
- [RFC] PyTorch Tensor Parallel(TP) User API for Distributed Training
- nn.CrossEntropy/nn.NLLLoss : Request for option to specify invalid ignore_index for perf. optimization
- [MPS] Add support for aten::repeat_interleave.self_Tensor for MPS backend
- DISABLED test_hf_bert_ddp_inductor (__main__.TestFakeDistributedSingleProc)
- addcmul on CUDA does not have the correct FMA behavior
- Docs
- Python not yet supported