optax.tree\_utils.tree\_cast
============================

.. currentmodule:: optax.tree_utils

.. autofunction:: tree_cast