private List<Integer> tokenize(String text) {
        var tokens = tokenizer.embed(text, new Embedder.Context("tokenizer"));
        tokens = tokens.size() >= tokenizerMaxTokens ? tokens.subList(0, tokenizerMaxTokens - 1) : tokens;
        tokens.add(TOKEN_EOS);
        return tokens;
    }
Given a tensor 'logits' with 3 dimensions: batch, sequence, and vocabulary
find the value in the vocabulary dimension with highest score for the given
token in the sequence
model-integration/src/main/java/ai/vespa/llm/clients/OnnxEncoderDecoder.java