updated docs

Author: srigas

docs/api-reference.rst

@@ -62,6 +62,28 @@ KAN Layer Classes
    :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
 --------------
 

docs/tutorials/Tutorial 11 - Optimizer Utilities.ipynb

@@ -119,7 +119,7 @@
    "id": "0e1983a1",
    "metadata": {},
    "source": [
-    "# Part 1: Adam Optimizer"
+    "## Part 1: Adam Optimizer"
    ]
   },
   {
@@ -135,7 +135,7 @@
    "id": "d90dbfe6",
    "metadata": {},
    "source": [
-    "## Experiment Setup for Adam"
+    "### Experiment Setup for Adam"
    ]
   },
   {
@@ -229,7 +229,7 @@
    "id": "5cbc2aee",
    "metadata": {},
    "source": [
-    "## Constant Learning Rate"
+    "### Constant Learning Rate"
    ]
   },
   {
@@ -280,7 +280,7 @@
    "id": "d800ac0a",
    "metadata": {},
    "source": [
-    "## Exponential Decay Schedule"
+    "### Exponential Decay Schedule"
    ]
   },
   {
@@ -340,7 +340,7 @@
    "id": "23133d54",
    "metadata": {},
    "source": [
-    "## Cosine Annealing Schedule"
+    "### Cosine Annealing Schedule"
    ]
   },
   {
@@ -393,7 +393,7 @@
    "id": "0e6da679",
    "metadata": {},
    "source": [
-    "## Warmup Strategy"
+    "### Warmup Strategy"
    ]
   },
   {
@@ -450,7 +450,7 @@
    "id": "2afa40c5",
    "metadata": {},
    "source": [
-    "# Part 2: L-BFGS Optimizer"
+    "## Part 2: L-BFGS Optimizer"
    ]
   },
   {
@@ -472,7 +472,7 @@
    "id": "32fb8201",
    "metadata": {},
    "source": [
-    "## Experiment Setup for L-BFGS"
+    "### Experiment Setup for L-BFGS"
    ]
   },
   {
@@ -575,7 +575,7 @@
    "id": "b53fd159",
    "metadata": {},
    "source": [
-    "## L-BFGS with Default Parameters"
+    "### L-BFGS with Default Parameters"
    ]
   },
   {
@@ -623,7 +623,7 @@
    "id": "8e6ae90c",
    "metadata": {},
    "source": [
-    "## L-BFGS with Larger Memory"
+    "### L-BFGS with Larger Memory"
    ]
   },
   {