ENH: generalize more functions to allow piecewise homogeneous Markov jump processes