Thank you for your hard work on this library! Do you have any plans to add StyleGAN(2)? There are unofficial numerous PyTorch implementations (e.g. https://github.com/rosinality/stylegan2-pytorch) on GitHub.