Well, it would make things very complicated and I don’t think they would go hand in hand. Well if your data loading process is becoming an issue, you can save intermediate tensors after applying transforms to disk and directly load them.