import { Callout } from "nextra-theme-docs" import { Tab, Tabs } from "nextra-theme-docs" VocabEmbedding Forward (x: jaxtyping.Integer[Tensor, '... n']) -> jaxtyping.Float[Tensor, '... n d']