Is autodifferentiation possible in this situation?

[ChrisRackauckas]

@xzackli

I’m guessing it’s because we turn off these tight coupling approximations when the evolution becomes only moderately stiff

This really sounds like a case for ROCK4. But yes, there’s are plenty of things to try which are not RK, and the papers you linked also recommended Radau which is there is multiple forms.