Here is the code used in the book - https://github.com/rasbt/LLMs-from-scratch