ParentFull threadwfalcon·this is also what our (Lightning AI) lit-gpt library does. https://github.com/Lightning-AI/lit-gptView on HN