Allow custom loss functions and optimization in Neural Network Predictor

[danyoungday]

Currently the Neural Network Predictor defaults to AdamW optimizer with L1Loss, however we would like to be able to customize this.