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

Module: tfp.experimental.nn.losses

Loss functions for neural networks.

Functions

compute_extra_loss(...)

kl_divergence_exact(...): Exact KL Divergence.

kl_divergence_monte_carlo(...): Monte Carlo KL Divergence.

negloglik(...): Negative log-likelihood.