optax.tree\_utils.tree\_set
===========================

.. currentmodule:: optax.tree_utils

.. autofunction:: tree_set