WIP: experimental support for rhs_with_state python function callbacks in mcsolve