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

[nnx] add qkv_features back to MHA #3566

cgarciae
cgarciae nnx: add qkv_features back to MHA
eb850d87
cgarciae cgarciae force pushed to eb850d87 2 years ago
codecov-commenter
cgarciae cgarciae changed the title nnx: add qkv_features back to MHA [nnx] add qkv_features back to MHA 2 years ago
chiamp
chiamp approved these changes on 2023-12-18
chiamp chiamp added pull ready
copybara-service copybara-service merged a59675c1 into main 2 years ago
copybara-service copybara-service deleted the nnx-attention-add-qkv-features branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone