A bit better tooling as well as better batching for training on mnist; notice how the matrices allow you to have variable numbers of rows allowing for training of multiple examples (1 per row) at once