rockpool.utilities.jax\_tree\_utils
===================================

.. automodule:: rockpool.utilities.jax_tree_utils

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
   
      tree_map_reduce_select
      tree_map_select
      tree_map_select_with_rng
      tree_map_with_rng