flax
added nnx attention equivalence test
#3687
Merged

added nnx attention equivalence test #3687

chiamp
chiamp chiamp assigned chiamp chiamp 1 year ago
chiamp chiamp requested a review from cgarciae cgarciae 1 year ago
codecov-commenter
chiamp chiamp force pushed 1 year ago
cgarciae
cgarciae approved these changes on 2024-02-10
chiamp chiamp added pull ready
chiamp added nnx attention equivalence test
6956a2d5
chiamp chiamp force pushed to 6956a2d5 1 year ago
copybara-service copybara-service merged a5661961 into main 1 year ago
chiamp chiamp deleted the nnx_attention branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone