jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
30k stars
334 watching
3k forks
Language: Python
last commit: 1 day ago
Linked from 6 awesome lists
jax
jax Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
josephmisiti/awesome-machine-learning
ethicalml/awesome-production-machine-learning
krzjoa/awesome-python-data-science
n2cholas/awesome-jax
amrzv/awesome-colab-notebooks
jonathandinu/ai4artists