Added NNX/Linen API consistency test for Linear/Dense layer

[Micky774]

What does this PR do?

Added NNX/Linen API consistency test for Linear/Dense layer

Fixes # (issue)

Checklist

• This PR fixes a minor issue (e.g.: typo or small bug) or improves the docs (you can dismiss the other
  checks if that's the case).
• This change is discussed in a Github issue/
  discussion (please add a
  link).
• The documentation and docstrings adhere to the
  documentation guidelines.
• This change includes necessary high-coverage tests.
  (No quality testing = no merge!)

[Micky774]

cc: @cgarciae

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (a572f6a) 53.80% compared to head (3024a14) 53.93%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3509      +/-   ##
==========================================
+ Coverage   53.80%   53.93%   +0.12%     
==========================================
  Files          98       98              
  Lines       11524    11546      +22     
==========================================
+ Hits         6201     6227      +26     
+ Misses       5323     5319       -4     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.