For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

pool_channel

def pool_channel[kpool: Int](logits: Array[Float32, kpool], vals: Array[Float32, kpool]) -> Float32

One pooled channel: softmax over logits, weighted sum of vals.

Shared by the prefill and decode writers so the two cannot drift.

Returns:

Float32