Skip to content

Add cross-attention to MultiheadAttention - #156

Merged
jessegrabowski merged 1 commit into
pymc-devs:mainfrom
jessegrabowski:cross-attention
Sep 5, 2026
Merged

Add cross-attention to MultiheadAttention#156
jessegrabowski merged 1 commit into
pymc-devs:mainfrom
jessegrabowski:cross-attention

Let MultiheadAttention read keys and values from a second input

3e2891a
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

1 warning
mypy
succeeded Sep 5, 2026 in 1m 8s