Code for (unfinished) multi gpu simclr blog