Func

func.jacrev

This function provides the exact same functionality as torch.func.jacrev(), except that it allows LieTensor to be used as input when calculating the jacobian.