flax
34452966 - [flax] Add QK-normalization to MultiHeadDotProductAttention

Commit

2 years ago

[flax] Add QK-normalization to MultiHeadDotProductAttention PiperOrigin-RevId: 556357734

Author

prazek

Committer

a-googler

Parents