optax.freeze
============

.. currentmodule:: optax

.. autofunction:: freeze