else:
       mps_device = torch.device("mps")
   # Create a Tensor directly on the mps device
   x = torch.ones(5, device=mps_device)
   # Or
   x = torch.ones(5, device="mps")

   # Any operation happens on the GPU
   y = x * 2

   # Move your model to mps just like any other device
   model = YourFavoriteNet()
   model.to(mps_device)

   # Now every call runs on the GPU
   pred = model(x)
