tfp.experimental.nn.losses.kl_divergence_monte_carlo

Monte Carlo KL Divergence.