Updated the docstring.
Improved the docstring of the derivatives of the losses.
Improved the citations.
Does not compile :<