Can we call this `state_derivatives()`? From what Tarun has said before, this is not actually going to be part of the code gen module, but maybe we can get started down the path of calling it `state_derivatives`.