flax

Add tests for nnx.initializers zeros_init and ones_init

#5276

Open