Train pix2pix on Google Cloud TPUs (based on @affinelayer's implementation) - View it on GitHub