Star AlbumentationsX on GitHub — it powers this leaderboard

← Back to leaderboard

jax-ml

jax

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

34,896stars

Homepage View on GitHub

Forks

3,424

Open issues

2,235

Watchers

34,896

Size

149.4 MB

PythonApache License 2.0

jax

Created: Oct 25, 2018

Updated: Feb 18, 2026

Last push: Feb 18, 2026