flax
11643b4b - Add in_kv_features argument to nnx.MultiHeadAttention, addressing https://github.com/google/flax/issues/4756.
Commit
View On GitHub
Commit
306 days ago
Add in_kv_features argument to nnx.MultiHeadAttention, addressing https://github.com/google/flax/issues/4756. PiperOrigin-RevId: 761681829
Author
Ryan McKenna
Committer
a-googler a-googler
Parents
a5eebe5d
FAQ Terms Privacy Refunds Impressum
Loading