Ajay and I discussed that a bit a while ago and there is a link to a blog post and Jupyter notebook doing the toy examples from the improved training article in pytorch:

  
    
    
    Wasserstein loss layer/criterion 
  
  
    Hi @AjayTalati, 
thanks for the pointers! I will definitely look into implementing an application or two. 
In the meantime, I jotted down a few thoughts regarding the Improved Training paper for your amusement while we’re waiting for the grad of grad to be merged.  Or maybe we find something to do with the original WGAN code, too. 
You have good holidays, 
best regards 
Thomas
  


Best regards
Thomas