jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.
What it solves
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more.
The project context on this page is free to read. The original GitHub repository remains the source of truth; sign in only when you want to save or join the discussion.
Best fit
Teams evaluating Developer Tools
Community notes