ENH: get_expm_augmented_tree will now accept edge-specific rate matrices