Repositories

jax-ml repositories

5 supported repositories

Show repositories with indexed issues
jax-ml/jaxPython

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Last commit Aug 19, 2026

 (36,180 stars) (3,738 forks) (4 indexed issues) (4 open good first issues)
jax-ml/jax-ai-stackPython

Last commit Aug 17, 2026

 (310 stars) (48 forks) (0 indexed issues) (0 open good first issues)
jax-ml/jax-blog

Last commit Jan 9, 2026

 (6 stars) (2 forks) (0 indexed issues) (0 open good first issues)
jax-ml/ml_dtypesC++

A stand-alone implementation of several NumPy dtype extensions used in machine learning.

Last commit Aug 19, 2026

 (357 stars) (60 forks) (0 indexed issues) (0 open good first issues)
jax-ml/oryxPython

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

Last commit Jul 8, 2026

 (320 stars) (13 forks) (1 indexed issue) (1 open good first issue)