Incidentally `diffrax.PIDController` also has a `dtmin` argument that could probably be used here instead of re-running things. :)