IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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).

Python module

max.nn.state_space.kimi_delta

Python wrapper for the KDA (Kimi Delta Attention) recurrence op.

Type aliases​

KdaBetaModeWhether the kernel applies the sigmoid to beta_logits itself.
KdaGateMode"original" is the stable-softplus gate; "safe" the bounded gate.
KdaStateLayout[N, HV, K, V] or [N, HV, V, K].

Constants​

KDA_GATE_LOWER_BOUNDHard-coded lower bound for gate_mode="safe".