Diffeq.jl v6.4: Full GPU ODEs, Neural ODEs with Batching on GPUs, and More
Wow, that's really cool. I feel like DiffEq is one of those libraries that seems to leverage every one of Julia's cool features that I can think of.<p>Making differential equations 'just work' on a GPU is super cool. I feel like if this were almost any other language, it'd be impossible to get this level of code-reuse where a user can just bring their own differential equation that may not even know about GPUs and then have the differential equation library solve it on the GPU by just giving a GPUarray as the initial condition.<p>________________________________<p>Also, since the blog post doesn't seem to link to it, here's a link to the actual DifferentialEquations.jl repo: <a href="https://github.com/JuliaDiffEq/DifferentialEquations.jl" rel="nofollow">https://github.com/JuliaDiffEq/DifferentialEquations.jl</a>