I would avoid repeating transformations and transforms the tensors once.
Often you also don’t need the same tensor in different dtypes or devices, so I’m wondering if e.g. offsets is used anywhere later on the CPU.