Gradient Surgery + Google colab GPUs