Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfp.experimental.mcmc.reconstruct_trajectories

Reconstructs the ancestor trajectory that generated each final particle.