Visualize, create, and operate on JAX PyTree in the most intuitive way possible.


Installation
------------

Install from pip::

   pip install pytreeclass


.. toctree::
   :caption: User guides
   :maxdepth: 1

   notebooks/getting_started
   notebooks/stateful_computations
   notebooks/dealing_with_non_jaxtype.ipynb
   notebooks/common_recipes

.. toctree::
   :caption: Examples
   :maxdepth: 1

   notebooks/build_mini_optimizer_library

.. toctree::
    :caption: API Documentation
    :maxdepth: 1
    
    API/pytreeclass

Apache2.0 License.

Indices
=======

* :ref:`genindex`