Very simple ML model (NN), purpose is to learn, as well as visualise,
Go here to visualise the 3D Loss model online!