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
- IValue::deepcopy(GenericDict) returns incorrect result when dictionary values are Tensor views
- Unnecessary compilation fails to optimize simple code
- aot-compile inference module, update_inactive_constant_buffer function, CUDA memory continues to increase
- [inductor][cpu]mobilenet_v3_large PTQ/QAT performance regression in 2024-05-04 nightly release
- Aborted (core dumped) in torch.fft.irfftn/hfftn/ihfftn with malloc(): corrupted top size
- [inductor][cpu]mobilenet_v2_quantized_qat float32 single thread static/dynamic shape CPP/default wrapper performance regression in 2024-04-28 nightly release
- [BUG]Nan in gradients of scaled_dot_product_attention operation with mem_efficient backend
- Unnecessary warning when numpy not installed
- DISABLED test_train_parity_2d_mlp (__main__.TestFullyShard2DTraining)
- result of 2 ** s differs between eager mode and inductor + triton + cuda when in float32 denormal range
- Docs
- Python not yet supported