API Documentation
=================

.. autosummary::
   :toctree: autosummary
   :template: custom-module-template.rst
   :recursive:

   galax.heterograph
   galax.function
   galax.core
   galax.data
   galax.nn