optax.masked
============

.. currentmodule:: optax

.. autofunction:: masked