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

4 points | by tucan9389 16 hours ago