The catch: JAX is a different programming model. Python loops become lax.fori_loop. Conditionals become lax.cond. You're writing functional array programs that happen to use Python syntax -- closer to a domain-specific language than a drop-in optimizer. But if your problem fits, the numbers speak for themselves. JAX isn't the only library that compiles array code -- PyTorch has torch.compile, for example. I only tested JAX, so I can't say whether others would produce similar results on these benchmarks.
There are welcome improvements in newer versions of this comparison, like the introduction of the “Format” panel on the right-hand side, which makes better use of widescreen landscape-oriented displays, and allows for larger controls. While I admire the density of the Lion-era screenshot, the mini-sized controls in that formatting menu are harder to click.1
，推荐阅读safew获取更多信息
Elsewhere, be sure to check out Storrie's undeniably impressive range of accents.
mentioned this pull request