Move data between CPU and GPU with DataModule

Is it possible to have a Lightning DataModule load a dataset on the CPU and then move single batches of that dataset to the GPU?

I have a large dataframe which fits in CPU RAM but not on GPU. I don’t really want to read batches from disk as this will be a lot slower than reading into RAM and subsequently moving batches to GPUs.