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 supported1216 Subscribers
Add a CodeTriage badge to pytorch
Help out
- Issues
- [Testcase Refactoring] Migrate test_fsdp_dtensor_state_dict to capability gating
- [Testcase Refactoring] Migrate test_fsdp_comm to capability gating
- `torch.linalg.svd` raises `RuntimeError` on MPS for batched inputs
- [CPP FakeTensor] to_list handling in cpp
- Distributed scatter/broadcast/reduce with Gloo backend return wrong results on non-contiguous tensors
- Stop mse_loss/smooth_l1_loss mean/sum reductions from retaining the full elementwise-loss storage
- [Testcase Refactoring] Add hardware classification to APoT quantizer tests
- [Testcase Refactoring] Add hardware classification to quantization utility tests
- `torch.linalg.norm` returns `inf` for norms well inside the dtype range, and its gradient is silently `0` (float32/float64)
- UNSTABLE rocm-navi31 / linux-jammy-rocm-py3.10-navi31 / test (default)
- Docs
- Python not yet supported