flax
Add NNX/Linen consistency test for `Embed` layer
#3513

Merged

Add NNX/Linen consistency test for `Embed` layer #3513

copybara-service merged 2 commits into google:main from Micky774:nnx_embed_api_test

Add NNX/Linen consistency test for `Embed` layer

9ad4d3f7

Merge branch 'main' into nnx_embed_api_test

b74e3a6e

cgarciae approved these changes on 2023-11-29

cgarciae added pull ready

copybara-service merged e50b1b53 into main 2 years ago

Micky774 deleted the nnx_embed_api_test branch 2 years ago

Reviewers

cgarciae

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone