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
173.4 MB
PythonApache License 2.0
Created: Apr 6, 2020
Updated: Jul 6, 2026
Last push: Jul 10, 2026
Fork