How to use adaptive loss function from google-research in Keras?

Question

Similar to this question, I am having some trouble using the adaptive loss function from robust loss (commit c24a2d8) from google-research: the AdaptiveLossFunction