flax
added nnx rmsnorm
#3691
Merged

added nnx rmsnorm #3691

copybara-service merged 1 commit into google:main from chiamp:nnx_rms
chiamp
chiamp chiamp assigned chiamp chiamp 2 years ago
chiamp chiamp requested a review from cgarciae cgarciae 2 years ago
codecov-commenter
cgarciae
cgarciae approved these changes on 2024-02-13
chiamp added nnx rmsnorm
172078e9
chiamp chiamp force pushed to 172078e9 2 years ago
chiamp chiamp added pull ready
copybara-service copybara-service merged c25f546f into main 2 years ago
chiamp chiamp deleted the nnx_rms branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone