It would nice to have the option to have a callback function `callback` passed as a keyword argument during training.