`before_batch_transfer` and `after_batch_transfer` hooks in LightningDataModule
🚀 Feature
Can we have two additional hooks in LightningDataModule?
Something like before_batch_transfer_to_device, after_batch_transfer_to_device, although these can be renamed to something else.
Motivation

before_batch_transfer_to_device: This can help apply transformations or augmentations to a batch before it is transferred to the device.
after_batch_transfer_to_device: This can help apply transformations or augmentations to a batch after it is transferred to the device. For eg we can perform data augmentation in the GPU for a batch using Kornia.

Alternatives
As of now, it can be done by overriding the transfer_batch_to_device hook but a separate hook makes more sense here I think.
If approved I can send a PR :)