1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 06:16:06 +00:00
2018-11-21 08:57:25 -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-21 08:57:25 -08:00
2018-11-19 20:18:31 -08:00

JAX is a research project that grew out of Autograd. Here's a two-page abstract about an early version. 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 152 MiB
Languages
Python 89.3%
C++ 8.1%
Starlark 1.5%
Jupyter Notebook 0.5%
Shell 0.3%
Other 0.3%