code: def embedding(ni:int,nf:int) -> nn.Module:
      "Create an embedding layer."
      emb = nn.Embedding(ni, nf)
      # See https://arxiv.org/abs/1711.09160
      with torch.no_grad(): trunc_normal_(emb.weight, std=0.01)
      return emf embedding(ni:int,nf:int) -> nn.Module:
      "Create an embedding layer."
      emb = nn.Embedding(ni, nf)
      # See https://arxiv.org/abs/1711.09160
      with torch.no_grad(): trunc_normal_(emb.weight, std=0.01)
      return emb
docs: Create an embedding layer.