- am-mixer-pack-batch samples seq-len hidden-dim dtypeprocedure
- am-mixer-unpack-batch data batch seq-len hidden-dim dtypeprocedure
Convert between a list of row-major (T, D) vectors, one per sample, and one token-major [T*B, D] typed vector.
The module also exports the building blocks make-am-token-mixing-mlp, make-am-channel-mixing-mlp and make-am-mlp-mixer-block, and the predicate am-mlp-mixer?.