Hi,
That looks like a bug in the max implementation for ByteTensor on cuda indeed. 
Could you open an issue on the pytorch main repo on github with this same reproducing example please?
Thanks for the report.