Machine Learning

478 readers

1 users here now

A community for posting things related to machine learning

Icon base by Lorc under CC BY 3.0 with modifications to add a gradient

founded 1 year ago

MODERATORS

Ategon@programming.dev

Akisamb@programming.dev

ericjmorey@programming.dev

GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch (programming.dev)

submitted 10 months ago by ericjmorey@programming.dev to c/machine_learning@programming.dev

1 comments fedilink hide all child comments

cross-posted from: https://programming.dev/post/8391233

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

you are viewing a single comment's thread
view the rest of the comments

[–] ericjmorey@programming.dev 1 points 10 months ago

Your submission in "GPU-based ODE solvers which are 20x-100x faster than those in #jax and #pytorch" was removed for Testing functions using new Proton UI.