updated rng docstring for init, apply and make_rng

[chiamp]

Updated docstring for Module methods init and apply to show examples of default RNG behavior. For more context, see:

• Make self.make_rng() callable by using a default RNG stream and have .init() and .apply() use the default RNG stream if no explicit RNG mapping is passed. #3668
• add default make_rng #3669
• make make_rng default to 'params' #3699
• added default params rng to .apply #3698

View the updated docstrings here: init, apply and make_rng.

[cgarciae]

Feels like we could also improve make_rng's docstring e.g. copy some of the examples / explanations from here.

[codecov-commenter]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 59.13%. Comparing base (718aa8c) to head (76f6c46).
Report is 6 commits behind head on main.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #3765   +/-   ##
=======================================
  Coverage   59.12%   59.13%           
=======================================
  Files         101      101           
  Lines       12468    12469    +1     
=======================================
+ Hits         7372     7373    +1     
  Misses       5096     5096           

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