jax-ml/jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

★ 36.6K Stars

⑂ 4.0K Forks

○ 36.6K Watchers

✖ 2.6K Issues

Python

jax

Last push: 2h ago

Created:

Loading additional information…