flax
[nnx] add qkv_features back to MHA
#3566
Merged

Loading