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