torch_max_mem
