How to train/predict a very simple feed-forward neural network in Flux?

@sylvaticus maybe my code will help:
Generic Function to train NN w/ Flux