gans

This needs to be done in 2 parts:

• Separate out the functional forms of the losses into a module of losses
• Document the necessary functions

We don't need to expose all the functions. Some functions do nothing fancy and they need to be removed and the entire computation can be performed inside the forward function.