Repositories

jax-ml repositories

4 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 Jun 7, 2026

 (35,767 stars) (3,610 forks) (0 indexed issues) (0 open good first issues)
jax-ml/jax-ai-stackPython

Last commit May 26, 2026

 (302 stars) (47 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 May 27, 2026

 (344 stars) (57 forks) (0 indexed issues) (0 open good first issues)