b3k / gpt-from-scratch

Build GPT from scratch using Nx and Axon

gpt-from-scratch

Reimplements the GPT model in Andrej Karpathy's video: https://www.youtube.com/watch?v=kCc8FmEb1nY using Nx/Axon/Livebook