Great to have an MNIST example. And the learning page is much more illustrative than before. I think we should consider two things:

• I appreciate removing the factual problems in the learning page, but I do think the current version is lacking some of the intuitive explanation I tried to provide before. The current page provides no introduction to spiking neurons, no explanation why we need surrogate gradients, and, in my opinion, not enough practical information on how to train SNNs with Norse. I think the details are great, but I would perhaps suggest keeping the (admittedly, really nice) visualization in a separate page about surrogate gradients. I would suggest the 'learning' page is a quick tour about how surrogate gradients work and how it's used in Norse.
• There are red warnings/errors in the MNIST dataset and it doesn't compile. How about only using a limited dataset and set epoch=1 as default to avoid build-time slowdown? We can always ask people to augment the data/epochs if they need. I doubt that'll interfere with the purpose of the notebook.