TF-GAN, A lightweight GAN library for TensorFlow 2.0


A newer version of TF-GAN (a lightweight library for training and evaluating Generative Adversarial Networks) was announced recently. TF-GAN has been used in a number of projects and papers 


Updated Features:

Cloud TPU support: TF-GANs can be used to train GAN’s Google Cloud TPU’s.

AdvertisementCoursera Plus banner featuring Johns Hopkins University, Google, and University of Michigan courses highlighting data science career-advancing content

Self-study GAN course: Open source self study GAN courses based on internal Google study materials.

PyPi package: TF-GAN can be installed with ‘pip install tensorflow-gan’ and used with ‘import tensorflow_gan as tfgan’.

GAN metrics: TF-GAN has easier metrics to compare results from papers. TF-GAN metrics are computationally-efficient and syntactically easy.

GitHub Repository: It has its own Github repository and can be accessed easily.

Colaboratory tutorials: TF-GAN can now used be used with Google GPU’s and TPU’s.

TensorFlow 2.0: TF-GAN is now compatible with TensorFlow 2.

Image source:

Projects using TF-GAN

In terms of projects, here are some below:

Self-Attention GAN on Cloud TPUs [Paper]

Image Extension [Paper]

BigGAN [Paper]

GANSynth [Source]




Please enter your comment!
Please enter your name here

This site uses Akismet to reduce spam. Learn how your comment data is processed.