You can assign where each part of the model is allocated manually. Moving tons of data would be efficient for training. As far as I know it should be complicated and it’s not designed to be done by default. I don’t know if there is a library to do that but I doubt it.