Nice showcase that interesting LLM work can be done on single

Press Space to continue

Nice showcase that interesting LLM work can be done on single GPU!

I extended the GPT-2-style code from @rasbt's "Build a Large Language Model (from Scratch)" so that it was a 6-expert (2 active) mixture-of-experts, and trained it from scratch over 8 days. It worked well! Full writeup with maths and code at https://t.co/7YsQifjWDk