SciML / DiffEqGPU.jl

GPU-acceleration routines for DifferentialEquations.jl and the broader SciML scientific machine learning ecosystem
https://docs.sciml.ai/DiffEqGPU/stable/
MIT License
272 stars 27 forks source link

Update to ForwardDiff support in EnsembleGPUKernel #278

Closed utkarsh530 closed 1 year ago

utkarsh530 commented 1 year ago

@ChrisRackauckas can you review? Also added a GPU kernel-compatible FiniteDiff version for differentiating through implicit ODE solvers.