junpenglao opened this issue 6 months ago · 1 comments
https://github.com/google-deepmind/optax/blob/main/examples/lbfgs.ipynb