This is an automated archive made by the Lemmit Bot.

The original was posted on /r/machinelearning by /u/internet_ham on 2024-11-02 13:10:48+00:00.

I love JAX, but I fully concede that you sacrifice ease of development for performance.

I’ve seen some buzz online about the speedups due to torch.compile, but I’m not really up to date. The is performance case for JAX dead now, or are the impressive GPU performance due to other factors like multi-GPU, etc.