Error in the lookahead optimizer example

[fabianp]

At the end of https://optax.readthedocs.io/en/latest/_collections/examples/lookahead_mnist.html there's a KeyboardInterrupt exception, probably from the example taking too long when generated from readthedocs