jax-ml/jax

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

★ 36.0K Stars

⑂ 3.7K Forks

○ 36.0K Watchers

✖ 2.5K Issues

Python

jax

Last push: 3h ago

Created:

Loading additional information…