Use BFGS optimizer in Tensorflow 2 almost as it was with Tensorflow 1 and tf.contrib

tf2-bfgs is a wrapper to train a tf.Module or tf.keras.Model using (L-)BFGS optimizer from Tensorflow Probability. This work uses the code snippet from Pi-Yueh Chuang available here.