Add vmap reset() regression test for nnx.metrics (#5483)
Construct a MultiMetric(loss=Average) under nnx.vmap so its state
carries a leading batch axis, call reset(), then run a vmapped update
and assert compute() returns the expected per-batch means. Locks in the
shape-preserving reset() behavior tracked in #5483.
