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