Clean up mnist runner
WIP: Implement classification via 'collision' layer

- Architecture would look like this: u -- U -- X -- Y -- y
- Also implement support for missing input values via nans