google/torchax
View on GitHubtorchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JAX-Pytorch interoperability, meaning, one can mix JAX & Pytorch syntax together when authoring ML programs, and run it in every hardware JAX can run.
- Stars
- 243
- Forks
- 38
- Open beginner issues
- 0
- Indexed issues
- 15
- Avg merge
- 15h 32m
- Merged PRs (30d)
- 2
- Dominant language
- Python
- License
- Apache-2.0
- Last GitHub push
- Sep 11, 2026
- Latest indexed
- Sep 16, 2026
- Contributing guide
- Contributing guide
- Code of conduct
- Code of conduct
- Beginner labels
- No beginner labels indexed
-
OperatorNotFound: Operator with name aten::_nested_tensor_from_mask_left_aligned has no lowering Open
google/torchax#12 · 3 comments · 0 reactions · 0 assignees ·
-
google/torchax#16 · 4 comments · 0 reactions · 0 assignees ·
-
google/torchax#22 · 13 comments · 0 reactions · 0 assignees ·
-
google/torchax#55 · 3 comments · 0 reactions · 0 assignees ·
-
google/torchax#62 · 2 comments · 0 reactions · 0 assignees ·
-
google/torchax#67 · 2 comments · 0 reactions · 0 assignees ·
-
google/torchax#105 · 0 comments · 0 reactions · 0 assignees ·
-
google/torchax#106 · 0 comments · 0 reactions · 0 assignees ·