• Chris Jewell's avatar
    Performance improvement · d4a78476
    Chris Jewell authored
    Re-wrote discrete_markov_log_prob to batch the call to
    tfd.Multinomial.log_prob to avoid lgamma time.
    d4a78476