Add reset_gate flag to MGUCell.

[carlosgmartin]

#3718

[codecov-commenter]

Codecov Report

Attention: Patch coverage is 25.00000% with 3 lines in your changes are missing coverage. Please review.

Project coverage is 58.74%. Comparing base (c2af614) to head (b7b7df2).

Files	Patch %	Lines
flax/linen/recurrent.py	25.00%	3 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3760      +/-   ##
==========================================
- Coverage   58.75%   58.74%   -0.02%     
==========================================
  Files         101      101              
  Lines       12413    12417       +4     
==========================================
+ Hits         7293     7294       +1     
- Misses       5120     5123       +3     

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

[cgarciae]

Thanks @carlosgmartin !

[chiamp]

Thanks!