STEM: Scaling Transformers with Embedding Modules | ResearchHub