Which one is the best implementation of LARS optimizer for pytorch?

I am doing research with self supervised learning. Currently most publications use LARS optimizer.
It seems there is no official PyTorch implementation.

So, which repository has the best (faster and reliable) implementation of LARS for PyTorch?

Thanks in advance.