@frgfm Just wanted to check in and ensure you are not blocked or need any assistance. Let me know if you do. :)

Hello there :)

My bad, I just caught up with the PR! I left it as a draft because the missing part is the unit test: I saw that the ones for the focal loss are quite extensive. Do I need to be that thorough?

@frgfm Ah yes the unit-tests. I would recommend following the pattern that we have on existing tests and have a similar coverage. Hopefully following the existing infra will make your work easier. The reason we really need them is because previously we spotted a few nasty bugs on some of the losses. We are only couple of weeks prior cutting the branch for the upcoming release so if we want to include this feature we need to have elevated degree of confidence.