# JAX

49 Hacker News commenters recommended JAX and 6 criticized it, November 2025 to November 2026. It was recommended in 12 of 13 months. It meets the Hacker Approved badge rule.

**Verdict:** Gives NumPy-style array code automatic differentiation and JIT compilation, with fast GPU execution through XLA. Common caveat: the debugging story is rough compared to PyTorch.

## Why people recommend it

- Fast GPU and compiled execution (5 comments): "Jax basically gives you the array language without leaving Python, and the XLA backend means you're not hand-tuning C for the GPU path." ([source](https://news.ycombinator.com/item?id=47386461)) "I've been moving to JAX after many years of using torch and XLA is kind of magic." ([source](https://news.ycombinator.com/item?id=46175566)) "PyMC uses jax under the hood now, so my MCMC workflows are GPU accelerated." ([source](https://news.ycombinator.com/item?id=46355085))
- Familiar NumPy-style API (4 comments): "Jax basically gives you the array language without leaving Python, and the XLA backend means you're not hand-tuning C for the GPU path." ([source](https://news.ycombinator.com/item?id=47386461)) "it's pretty easy to switch to Jax's NumPy API, and then you can easily jit in Python as well" ([source](https://news.ycombinator.com/item?id=47178496)) "If you're happy with NumPy's API, then surely JAX is exactly what you're looking for." ([source](https://news.ycombinator.com/item?id=48070160))
- Good for Bayesian and ML models (4 comments): "Thanks to efficient variational Bayes methods built on top of JAX, it is doable even on high-dimensional models." ([source](https://news.ycombinator.com/item?id=47473603)) "Go with Jax if you want a PyTorch alternative that can have better performance for certain scenarios." ([source](https://news.ycombinator.com/item?id=47206134)) "PyMC uses jax under the hood now, so my MCMC workflows are GPU accelerated." ([source](https://news.ycombinator.com/item?id=46355085))
- Autodiff and JIT compilation (3 comments): "it's pretty easy to switch to Jax's NumPy API, and then you can easily jit in Python as well" ([source](https://news.ycombinator.com/item?id=47178496)) "From a compiler standpoint JAX is much easier to work with since its all pure functions." ([source](https://news.ycombinator.com/item?id=50179534)) "using JAX for JIT-compilation and automatic differentiation" ([source](https://news.ycombinator.com/item?id=48298307))

## What people warn about

- Weak debugging and editor feedback (2 comments): "the debugging story is still rough compared to PyTorch" ([source](https://news.ycombinator.com/item?id=48192036)) "you don't get feedback as you edit" ([source](https://news.ycombinator.com/item?id=48174604))
- Awkward for some code patterns (2 comments): "Most of it can be expressed well with JAX's programming model, but the last 10% really sucks." ([source](https://news.ycombinator.com/item?id=48909501)) "It's less natural if your basic unit is an element of a vector." ([source](https://news.ycombinator.com/item?id=45846463))
- Dynamic shapes and Numba gaps (2 comments): "JAX can’t do what Numba can do for example." ([source](https://news.ycombinator.com/item?id=48070602)) "dynamic shapes were still an issue" ([source](https://news.ycombinator.com/item?id=47348182))

## Alternatives hackers recommend

- [PyTorch](https://hackerapproved.com/p/pytorch.md): 131 recommend
- [MLX](https://hackerapproved.com/p/mlx.md): 75 recommend
- [OpenCV](https://hackerapproved.com/p/opencv.md): 46 recommend
- [tinygrad](https://hackerapproved.com/p/tinygrad.md): 40 recommend
- [TensorFlow](https://hackerapproved.com/p/tensorflow.md): 20 recommend

## Cite as

Hacker Approved, "JAX", https://hackerapproved.com/p/jax/ (counts from 1,795,640 Hacker News comments, November 2025 to November 2026).