RetroFed Forum
Your home away from modern life.
GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch
Post Reply
AuthorMessage
ericjmorey programming.dev ericjmorey@programming.dev
Joined: 2023-06-17
Quote

Dr. Chris Rackauckas
(@chrisrackauckas@fosstodon.org) 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   Original
Post Reply
Back to top