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 supported919 Subscribers
View all SubscribersAdd a CodeTriage badge to pytorch
Help out
- Issues
- Performance for templated attention strangely bad for a combination of causal + upper square mask
- Pipeline Parallelism support for ZeroBubble
- Error when running DDP examples to train between nodes
- llvm disabled in official builds from 2.1.0 onwards
- [ROCm][CI] upgrade CI to ROCm 6.1
- [ROCm] hipSPARSELt Integration
- PreDispatch tracing doesn't produce correct "val" for Sparse Fake tensors.
- Allow tensor subclasses and add `torch.serialization.mark_safe_globals` that allows users to allowlist classes for `weights_only` load
- [inductor] switch assume_aligned_inputs to False
- Dynamo Export: Support for PixelShuffle
- Docs
- Python not yet supported