flax
9ba968b4
- [nnx] add more decode flag options for MultiHeadAttention
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[nnx] add more decode flag options for MultiHeadAttention
References
nnx-mha-decode
Author
cgarciae
Committer
cgarciae
Parents
7cbc3c13
Loading
