flax

Migrate `metrics.py` from `.value` to `[...]`

#4839

Open