- am-next-token-targets sequences vocab-size dtypeprocedure
For sequences of length T+1, returns a list (inputs targets) of one-hot typed vectors: the inputs are the first T tokens of each sequence, and the targets the last T.
The module also exports the building blocks make-am-attention, make-am-swiglu-ffn and make-am-transformer-block, and the predicate am-transformer?.