search

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

`EnsembleGPUArray` compatibility with different time intervals #283

Closed utkarsh530 closed 1 year ago

utkarsh530 commented 1 year ago

Could be fixed with this reparameterization trick: https://github.com/rtqichen/torchdiffeq/issues/122#issuecomment-738978844