def _convert_counts_logits_to_mean_disp(total_count, logits):
    """NB parameterizations conversion  - Copied over from scVI
    :param total_count: Number of failures until the experiment is stopped.
    :param logits: success logits.
    :return: the mean and inverse overdispersion of the NB distribution.
    """
    theta = total_count
    mu = logits.exp() * theta
    return mu, theta