That will certainly help as I convert the code over to use the GPU.  It’s amazing how much effort seems to go into making sure tensors are in the correct place and not contributing to GPU memory when they don’t need to be.  I’m sure there’s a good reason for this that I just don’t understand yet, but from a newbie’s perspective it would be nice if most of that were automated for you.