flax
Add `attention_bias` parameter to `MultiHeadDotProductAttention`.
#4694
Merged
Merged