.. _apiref:

API Reference
=============


KAN Layer Classes
-----------------

.. automodule:: jaxkan.layers.__init__
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Spline
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Chebyshev
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Legendre
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Fourier
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.RBF
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Sine
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.layers.Dense
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.layers.utils
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__


KAN Grid Classes
-----------------

.. automodule:: jaxkan.grids.BaseGrid
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__
   
.. automodule:: jaxkan.grids.SplineGrid
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.grids.RBFGrid
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__


KAN Models
--------------

.. automodule:: jaxkan.models.KAN
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.models.RGAKAN
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.models.ActNet
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.models.KKAN
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.models.utils
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__


PIKAN Training
--------------

.. automodule:: jaxkan.pikan.pde
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.pikan.sampling
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.pikan.adaptive
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__

.. automodule:: jaxkan.pikan.utils
   :members:
   :undoc-members:
   :show-inheritance:
   :special-members: __init__, __call__