flax
a572f6af
- Merge pull request #3487 from google:nnx-mha
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Merge pull request #3487 from google:nnx-mha PiperOrigin-RevId: 586134147
References
#3487 - [nnx] Add LinearGeneral and MultiHeadAttention
Author
a-googler
Parents
632e6c0a
0a1f78a3
Loading
