modded-nanogpt world record: Decoupling embedding size from model dimension

I have achieved a (not yet official) modded-nanogpt medium track world record.