A simple model for testing reasoning on a test dataset.
This model is trained 80 steps (toally 5 epochs)