jax
jakevdp/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
10stars
Forks
3
Open issues
0
Watchers
10
Size
186.7 MB
PythonApache License 2.0
Created: Apr 6, 2020
Updated: Jul 6, 2026
Last push: Oct 6, 2026
Fork