jax

jax

32982
3116

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

Community

GitHub

Languages

Python
86%
C++
11.9%
Starlark
1.8%
Shell
0.3%

Good First Issues

Ideal for first-time contributors

All Issues

Browse every open issue in this repository

README.md