jax-ml/jax

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

Python35,669 stars3,587 forksApache-2.0

jax

Rankings

Daily

#1093

Weekly

#1496

Monthly

#1461

Total stars

Daily new stars

Get badge

This repository is tracked by Trending Repos. The badge upgrades automatically if it ever cracks the top 100.

Markdown

![Trending Repos](https://trending-repos.com/badge/jax-ml/jax.svg)

HTML

<img src="https://trending-repos.com/badge/jax-ml/jax.svg" alt="Trending Repos" />

Raw URL

https://trending-repos.com/badge/jax-ml/jax.svg