Giters

jax-ml / jax

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

Home Page:http://jax.readthedocs.io/

Repository from Github https://github.comjax-ml/jaxRepository from Github https://github.comjax-ml/jax

jax-ml/jax Stargazers