This is really cool! Awesome progress :).
As a couple of thoughts, perhaps this would be ready to make a first PR to DeepChem with the implementation for 1st round code-review. Typically we’ll ask for some more unit tests and other modifications in the review process so might be good to kick that process off sooner rather than later. We might be able to discuss some of the changes to GAN/WGAN you suggest on that PR and kick off some fixes