Rename loss -> stochasticLoss and add a note to explain why we have
multiple train methods.