Supercharged high-resolution ocean simulation with Jax

Re:Supercharged high-resolution ocean simulation with Jax by fault1 on Monday December 06, 2021 @11:07AM (#29461452)

I'm also surprised that XLA.jl doesn't seem to have had continued development: https://github.com/FluxML/XLA.jl

When in doubt, piggybacking on (or at least interoperating with) what the large technology companies are investing in is probably savvy, sort of what the OP did.

[ Parent | on HN ]

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 ]