replaced zeros and ones initializers with zeros_init and ones_init builder functions

[chiamp]

refactoring the code base to use zeros_init and ones_init, as a follow up from #2790.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[codecov-commenter]

Codecov Report

Merging #2807 (88046e5) into main (e51d017) will not change coverage.
The diff coverage is 100.00%.

@@           Coverage Diff           @@
##             main    #2807   +/-   ##
=======================================
  Coverage   81.23%   81.23%           
=======================================
  Files          53       53           
  Lines        5659     5659           
=======================================
  Hits         4597     4597           
  Misses       1062     1062           

Impacted Files	Coverage Δ
flax/core/lift.py	96.06% <ø> (ø)
flax/core/nn/attention.py	13.47% <ø> (ø)
flax/core/nn/linear.py	42.85% <ø> (ø)
flax/core/nn/normalization.py	46.66% <ø> (ø)
flax/errors.py	85.58% <ø> (ø)
flax/linen/__init__.py	100.00% <ø> (ø)
flax/linen/transforms.py	94.14% <ø> (ø)
flax/linen/attention.py	95.96% <100.00%> (ø)
flax/linen/linear.py	97.51% <100.00%> (ø)
flax/linen/recurrent.py	100.00% <100.00%> (ø)

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.