- It is now possible to override the `to_device` method used to transfer model inputs and outputs between devices (fixes 4) - The parallel implementation of the `predict` method has been removed as it introduced needless challenges
2.1.9
- Learning Rate Callbacks can now be passed a metric in the `step` function. This enables the usage of the `ReduceLROnPlateau` learning rate scheduler.