mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
Description
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Languages
Python
89.3%
C++
8.1%
Starlark
1.5%
Jupyter Notebook
0.5%
Shell
0.3%
Other
0.3%