=================================
API for individual stages of HPS
=================================


Local Solve Stage
-----------------

.. automodule:: jaxhps.local_solve
   :members:

Merge Stage
------------

.. automodule:: jaxhps.merge
   :members:

Upward Pass 
----------------
.. automodule:: jaxhps.up_pass
   :members:

Downward Pass
----------------

.. automodule:: jaxhps.down_pass
   :members: