jax-ml/jax
View on GitHub ↗ ★ 36,567 Python
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
82 mentions on Hacker News by 61 people: 11 stories, 71 comments · first 2018-11-21 · latest 2026-04-05
20082026
-
▲ 3
jaxstory by tosh · 2026-04-05 · 0 comments
-
▲ 67
Show HN: Localscope–Limit scope of Python functions for reproducible executionstory by tillahoffmann · 2025-03-17 · 31 comments
-
▲ 4
Jax – Composable transformations: differentiate, vectorize, JIT to TPUstory by graderjs · 2022-03-21 · 0 comments
-
▲ 3
JAX and Haskellstory by unignorant · 2021-08-27 · 0 comments
-
▲ 9
Jaxstory by tosh · 2021-07-04 · 0 comments
-
▲ 1
How to Tame Your Jax: Navigating TPU VMs with TFRC in 2021story by sillysaurusx · 2021-06-23 · 2 comments
-
▲ 3
How to Tame Your Jax: Navigating TPU VMs with TFRC in 2021story by sillysaurusx · 2021-06-22 · 4 comments
-
▲ 235
Jax – Composable transformations of Python and NumPy programsstory by lelf · 2020-04-08 · 65 comments
-
▲ 2
JAX now runs on Cloud TPUs (with example collab notebooks)story by ArtWomb · 2019-12-15 · 0 comments
-
▲ 132
JAX: Numpy with Gradients, GPUs and TPUsstory by one-more-minute · 2018-12-08 · 29 comments
-
▲ 3
Google jaxstory by sahin-boydas · 2018-11-22 · 0 comments