pytree_utils
======================

.. py:module:: jaxns.internals.pytree_utils

.. rubric:: :code:`jaxns.internals.pytree_utils`


.. rubric:: Module Contents

.. py:function:: tree_dot(x, y)

.. py:function:: tree_norm(x)

.. py:function:: tree_mul(x, y)

.. py:function:: tree_sub(x, y)

.. py:function:: tree_div(x, y)