NewInference will return a new inference object with the given trained weights file to handle predictions.

Train will train a mnist network for given dataset object.

Inference is the internal representation of the inference object.