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