netket.jax.tree\_size
=====================

.. currentmodule:: netket.jax

.. autofunction:: tree_size