This is great work, I could not believe the fact that it keeps state (like the current epoch number, and of course all the current model state). Makes Python ev

This is great work, I could not believe the fact that it keeps state (like the current epoch number, and of course all the current model state). Makes Python even more dynamic. Nice that you could condense it to less than 100 lines of Python!