In my case, I greatly admire what folks at Huggingface have built in the candle[0] project. The project itself is amazing. I haven't yet delved deeper into the source code, but the code in their examples is clear and easy to understand

[0] https://github.com/huggingface/candle