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 21, 2026

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

Last commit Aug 17, 2026

 (311 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 20, 2026

 (358 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)