Home / JAX / Distrax Distrax Reimplementation of TensorFlow Probability, containing probability distributions and bijectors. Package 627 stars GitHub Back to JAX