Porting nanochat to a TPU: what carries over from PyTorch, and what breaks

For people that are like me : This entire text is AI generated, i feel weird reading it personally, i guess others may not

Nice work. Regarding your first insight, that's a sneaky one ^^. I experiment a similar issue with fine-tuning a small model with MLX. It was very hard to find out.

[flagged]