flax

Ignore Orbax warning in deprecated `flax.training.checkpoints.py` to unbreak head doctest

#4092

Merged