flax

6f745922 - fix MultiMetric GraphDef not hashable under jit (#5489)

fix MultiMetric GraphDef not hashable under jit (#5489) Store _metric_names as tuple instead of list so the static GraphDef produced by nnx.split() is hashable when passed to jax.jit/nnx.jit.