- [ ] Avoid string based lookups - [x] Scale embeddings once after vocab read and use the scaled vocab.