-
|
I've been using flax for a while and it's the best neural network module available as of now, to say the least. My question is short: Thank you in advance. |
Beta Was this translation helpful? Give feedback.
Answered by
IvyZX
Oct 11, 2024
Replies: 1 comment 1 reply
-
|
Nvm I saw it in the nnx source. |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Yes we simplified the name to
MultiHeadAttention. Please check out our API reference for how to use it.