astrobdr

I would really like to have such a feature! Please let me know if you are able to find a way to integrate this

Yes! Happy to help out! I can make some beginner notebooks or something. Any direction for where to start? @williamFalcon Ive made a few small contributions to lightning in the...

Hey @williamFalcon @lantiga Ive made exactly the code I was looking for in https://github.com/Lightning-AI/litgpt/pull/1331 :)