flax
Add in_kv_features argument to nnx.MultiHeadAttention, addressing https://github.com/google/flax/issues/4756.
#4757
Merged
Overview
Commits
1
Changes
View On GitHub

Add in_kv_features argument to nnx.MultiHeadAttention, addressing https://github.com/google/flax/issues/4756. #4757

copybara-service merged 1 commit into main from test_760691901
copybara-service
copybara-service copybara-service force pushed from 9aa6040c to 8be08065 313 days ago
copybara-service copybara-service force pushed from 8be08065 to e942aeba 310 days ago
Add in_kv_features argument to nnx.MultiHeadAttention, addressing htt…
11643b4b
copybara-service copybara-service force pushed from e942aeba to 11643b4b 310 days ago
copybara-service copybara-service merged 11643b4b into main 310 days ago
copybara-service copybara-service deleted the test_760691901 branch 310 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum