Hi all, I’m running into a problem when trying to compute gradients of an ODE solution on GPU (with Metal.jl) using SciMLSensitivity ( SciMLSensitivity: Automatic Differentiation and Adjoints for (Differential) Equation Solvers · SciMLSensitivity.jl ). Here’s a minimal working example: using Random…

oh, aparently you need to set abstol and reltol as Float32 also. We should make those be of utype also… Edit: it looks like we already do this. I’m very confused as to what’s going wrong…

SciMLSensitivity fails on GPU?

Specific Domains Machine Learning

Oscar_Smith September 22, 2025, 8:30pm 5

One thing worth asking is are you using up to date package versions?

1 Like

Topic		Replies	Views
Trouble with NeuralODE and Metal Numerics question , sciml	3	461	December 28, 2023
DiffEqGPU.jl with CUDA: Error computing gradients through SDE solver Machine Learning	12	352	January 10, 2024
Impossible to perform autodiff on simple ODE Optimization (Mathematical) ode , zygote , autodiff , differentialequation	22	593	November 1, 2024
Is this impossible for Zygote Adjoint to handle General Usage zygote	4	628	April 7, 2021
sciML_train func running but not optimizing New to Julia diffeq , flux , optimization , sciml	5	563	April 14, 2021

SciMLSensitivity fails on GPU?

Related topics