I was looking for something similar today. I just adopted it. :)

Thank you for your amazing work!

Do you have the intention to integrate it with PT Lightning or as a PT Lightning Logger?

It would be nice to have it maintained there.

It’s used all over huggingface-Transformers examples.