flax
Add `attention_bias` parameter to `MultiHeadDotProductAttention`.
#4701
Open

Loading