Nice work. I really appreciate this tool for enhancing my limited understanding the mechanism(s) behind attention and LLMs.