Remove flax optim test now that flax.optim is deprecated in favor of optax https://github.com/google/flax/issues/2273
Remove flax optim test now that flax.optim is deprecated in favor of optax https://github.com/google/flax/issues/2273

PiperOrigin-RevId: 477787392

google/fedjax,google/fedjax