Trying to code the paper "Memorizing Transformers" with the help of ChrisMcCormickAI's video series and lucidrains's codebase