ENH: two stage decomposition for faster expm when a rate matrix is updated in a block-structured way