flax

Update orbax_upgrade_guide.rst fixing broken link

#3031

Merged