flax
eb850d87 - nnx: add qkv_features back to MHA

Commit
2 years ago
nnx: add qkv_features back to MHA
Author
Committer
Parents
Loading