Why hackers moved from PyTorch to JAX

1 hacker described this switch in public comments. Their words, with links to the originals.

PyTorch 0 hackers criticize it
→
JAX 1 hacker recommends it

What hackers said about the move

“I switched from pytorch to jax just before triton appeared.”
sega_sai, 2025-11-07original comment ↗

Use PyTorch? Get alerted when hackers criticize it.

Tell us the tools in your stack. When hackers start criticizing one of them, or moving to an alternative, you will hear about it first.

PyTorch vs JAX

PyTorchJAX
Hackers who recommend11
Hackers who criticize00
Moved away from it10
Moved to it01

Counts are distinct hackers in the sampled comments, not market share.

Criticism of PyTorch

No further criticism of PyTorch captured yet.

Praise for JAX

No further praise for JAX captured yet.