@alisiafadini @minhuanli we should write a little documentation (simple & short!) about how to add a new loss I would vote it goes straight into the `README.md`.