code: def negative_binomial_dist(
      concentration, probs=None, *, logits=None, overdispersion=0.0
  ):
      _validate_overdispersion(overdispersion)
      if _is_zero(overdispersion):
          return dist.NegativeBinomial(
              concentration, probs=probs, logits=logits
          )
      raise NotImplementedError(
          "TODO return a NegativeBinomial or GammaPoisson"
      )
  
file_path: pyro/pyro/contrib/epidemiology/distributions.py
imports: import math
  from contextlib import contextmanager
  import torch
  from torch.distributions.utils import broadcast_all
  import pyro.distributions
  from pyro.distributions.util import is_validation_enabled
label_desc: Use image node_id 12 for calling a global function with example usage: negative_binomial_dist(concentration, probs) and returns: dist