Hi,

Yes, it was implemented on top of TensorFlow Probability sts package so it's fully Bayesian (that's why results are equivalent to the original package).

There are two methods for solving the posterior, variational inference (faster and less accurate) and hamiltonian monte carlo (slower and quite accurate).