ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.experimental.marginalize

Marginalizable probability distributions.

Classes

class MarginalizableJointDistributionCoroutine: Joint distribution parameterized by a distribution-making generator.

Functions

logeinsumexp(...): Computing einsum in logarithmic space.