2018-11-21 06:04:06 -08:00
2018-11-19 20:18:31 -08:00
2018-11-19 21:31:39 -08:00
2018-11-19 21:58:15 -08:00
2018-11-19 21:31:39 -08:00
2018-11-18 15:17:53 -08:00
2018-11-19 20:18:31 -08:00

JAX is a research project that grew out of Autograd. Watch this space for updates!

This is not an official Google product.

Description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Readme Apache-2.0 146 MiB
Languages
Python 89.3%
C++ 8.1%
Starlark 1.5%
Jupyter Notebook 0.5%
Shell 0.3%
Other 0.3%