flax
9ad4d3f7 - Add NNX/Linen consistency test for `Embed` layer

Commit

2 years ago

Add NNX/Linen consistency test for `Embed` layer

References

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

Author

Micky774

Micky774

Parents

Loading