Counting NanoGPT FLOPs in PyTorch
domluna.com