jax

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

30k stars

336 watching

3k forks

Language: Python

last commit: 8 days ago

Linked from 4 awesome lists

jax

Screenshot of jax-ml/jax website

http://jax.readthedocs.io/

Backlinks from these awesome lists: