Hi!
We do not support unified memory in pytorch.
There are only very narrow use case where it brings big improvements in practice unfortunately.
Is there any specific application you have that requires it?