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
149.8 MB
PythonApache License 2.0
Created: Apr 6, 2020
Updated: Feb 21, 2025
Last push: Feb 27, 2026
Fork