netket.jax.tree\_ax
===================

.. currentmodule:: netket.jax

.. autofunction:: tree_ax