``braintools.tree`` module
==========================

.. currentmodule:: braintools.tree 
.. automodule:: braintools.tree 

Operations for manipulating JAX PyTrees, including arithmetic, reshaping, and
conversion utilities used across the toolkit.

.. autosummary::
   :toctree: generated/

   scale
   mul
   shift
   add
   sub
   dot
   sum
   squared_norm
   concat
   split
   idx
   expand
   take
   as_numpy