Issue with PDMP and Forwardiff - DifferentialEquation

It looks like you need a call to reshape in the way you’re doing the dual cache? I’d have to dig in to find out why your integration is in 3D but the cache is 2D