flax
added nnx attention equivalence test
#3687
Merged

added nnx attention equivalence test #3687

chiamp
chiamp chiamp assigned chiamp chiamp 2 years ago
chiamp chiamp requested a review from cgarciae cgarciae 2 years ago
codecov-commenter
chiamp chiamp force pushed 2 years 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 2 years ago
copybara-service copybara-service merged a5661961 into main 2 years ago
chiamp chiamp deleted the nnx_attention branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone