GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch
GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch
![](https://programming.dev/pictrs/image/01716c58-ce65-47da-ba04-7250404ce5ad.png?format=webp&thumbnail=128)
![](https://programming.dev/pictrs/image/01716c58-ce65-47da-ba04-7250404ce5ad.png?format=webp)
cross-posted from: https://programming.dev/post/8391233
Dr. Chris Rackauckas (@[email protected]) writes:
#julialang GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch? Check out the paper on how #sciml DiffEqGPU.jl works. Instead of relying on high level array intrinsics that #machinelearning libraries use, it uses a direct kernel generation approach to greatly reduce the overhead.
Read Automated translation and accelerated solving of differential equations on multiple GPU platforms
1 comments