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 supported925 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- Allow positional arguments to be passed as kwargs for autograd custom Function
- Option to let DistributedDataParallel know in advance unused parameters at each forward pass
- Couldn't install pytorch 2.0
- forward-mode AD formula for torch.add (and possibly others) accidentally upcasts float32 to float64
- [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
- Basic math operations produce a "floating point exception"
- third-order gradient of torch.pow with tensor args and certain input returns NaN
- torch.addbmm throws different exception differences on CPU and GPU.
- Segfault on torch.nn.functional.one_hot with large tensor on Python 3.9
- Docs
- Python not yet supported