flax
96804855 - Add `attention_bias` parameter to `MultiHeadDotProductAttention`.

Commit

333 days ago

Add `attention_bias` parameter to `MultiHeadDotProductAttention`. PiperOrigin-RevId: 747024381

References

#4701 - Add `attention_bias` parameter to `MultiHeadDotProductAttention`.

Author

Jan Hosang

Committer

a-googler

a-googler

Parents

Loading