iShohei220

In this fixed? I'm still struggling to use SGLD with tf.function. ``` @tf.function def train(): optimizer = tfp.optimizer.StochasticGradientLangevinDynamics( learning_rate=1e-3, ) for epoch in range(15): for x, y in train_dataset: with...

https://github.com/tensorflow/probability/blob/v0.10.0/tensorflow_probability/python/optimizer/sgld.py#L273-L280 I read the implementation of SGLD, and understand why eager mode is not supported for SGLD. But based on the original pSGLD paper, the calculation of `preconditioner_grads` (Gamma in...

@googlebot I signed it!