News for hackers. Stuff that matters.

A comment on Supercharged high-resolution ocean simulation with Jax. Parent comment

Re:Supercharged high-resolution ocean simulation with Jax by adgjlsfhk1 on Monday December 06, 2021 @01:10PM (#29463411)
XLA.jl was kind of a solution looking for a problem. If you want fast code in Julia, you can just write Julia.
[ Parent | on HN ]
Re:Supercharged high-resolution ocean simulation with Jax by PhilipVinc on Monday December 06, 2021 @04:35PM (#29466019)
That's incorrect. If you work with mid-sized neural networks and MCMC sampling, allocations start to play a significant role (And Flux.jl is bad at preallocation). Prealloc.jl does not work properly. Zygote.jl adds even more allocations to the mix...

Jax/XLA completely solves this problem. Yes, it's annoying that you have to work with a static graph but if your problem fits the description... it's great.

[ Parent | on HN ]
Re:Supercharged high-resolution ocean simulation with Jax by dklend122 on Monday December 06, 2021 @05:12PM (#29466364)
There's work being done to solve this in Julia. See escapeanalysis.jl and the immutable array pr in base
[ Parent | on HN ]