4 ms·

Here's another simple implementation: https://github.com/karpathy/minGPT https://github.com/karpathy/minGPT