Updated API Reference for docs

Author: srigas

docs/api-reference.rst

@@ -48,6 +48,12 @@ KAN Layer Classes
    :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:
@@ -65,6 +71,24 @@ KAN Models
    :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: