jax-ml

jax-ml/jax

steady

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

Python View on GitHub

Stars

36,338

Forks

3,808

Open issues

1,764

24h

+6

+0.0%

7d

+39

+0.1%

Refresh

1h

Star history (7 days)

Last checked

13m ago

Last pushed

16m ago

Next check

just now