Fix `masked` wrapper for empty params nest.
Fix `masked` wrapper for empty params nest.

Currently passing an empty params nest to `optax.adamw(..., mask=<something>).update` fails with a ValueError (transform.py:565) although this should be valid.

PiperOrigin-RevId: 379242894

deepmind/optax,deepmind/optax