Add a setup to completely train a tiny model on made-up data. This can be used to test all implementations.