jaxsnn.event.custom\_lax
========================





.. automodule:: jaxsnn.event.custom_lax

   
   
   

   
   
   

   
   
   

   
   
   .. rubric:: Functions

   
   .. autofunction:: cond
   .. autofunction:: scan
   .. autofunction:: tree_flatten
   .. autofunction:: tree_map
   .. autofunction:: tree_unflatten