Remove TF dependency in `Lookahead Optimizer on MNIST` example #1568

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Open

rajasekharporeddy wants to merge 1 commit into google-deepmind:main from rajasekharporeddy:lookahead_mnist

Collaborator

rajasekharporeddy commented Jan 22, 2026

This PR removes TensorFlow as a dependency in lookahead_mnist.ipynb, and uses only TFDS, with grain for loading and preparing the dataset.


          Remove TF dependency in Lookahead Optimizer on MNIST example

2f590f7

vroulet approved these changes

View reviewed changes

Collaborator

vroulet left a comment

Looks great thanks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet