flax

948ee260 - Removes flax.deprecated.nn.

Removes flax.deprecated.nn. Removes flax.deprecated.nn. Also updates some references from flax.deprecated.nn to flax.linen if possible. PiperOrigin-RevId: 424339883