jax

jax

32531
3057

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

Community

GitHub

Languages

Python
86%
C++
12%
Starlark
1.7%
Shell
0.3%

README.md

Good First Issues

Ideal for first-time contributors

All Issues

Browse every open issue in this repository