Seems like nnclr would be covered also.

https://arxiv.org/pdf/2104.14548.pdf