Is it just obtaining a distribution of the next token predictions, or is it more complex?