Update readme

Author: BirkhoffG

README.md

@@ -8,9 +8,12 @@ status](https://github.com/BirkhoffG/explainax/actions/workflows/test.yaml/badge
 ![pypi](https://img.shields.io/pypi/v/explainax.svg) ![GitHub
 License](https://img.shields.io/github/license/BirkhoffG/explainax.svg)
 
+[**Installation**](#installation) \|
+[**Documentation**](https://birkhoffg.github.io/explainax/)
+
 `explainax` is a JAX library for model explanation and interpretation.
 
-## Quick install
+## Installation
 
 Simply install via `pypi`:
 

nbs/index.ipynb

@@ -14,7 +14,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Explainax: An Explainable AI Library for JAX\n"
+    "# Explainax: An Explainable AI Library for JAX"
    ]
   },
   {
@@ -29,7 +29,14 @@
     "![pypi](https://img.shields.io/pypi/v/explainax.svg) \n",
     "![GitHub License](https://img.shields.io/github/license/BirkhoffG/explainax.svg)\n",
     "\n",
-    "\n",
+    "[**Installation**](#installation) \\|\n",
+    "[**Documentation**](https://birkhoffg.github.io/explainax/) \n"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
     "`explainax` is a JAX library for model explanation and interpretation."
    ]
   },
@@ -38,7 +45,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "## Quick install\n",
+    "## Installation\n",
     "\n",
     "Simply install via `pypi`:\n",
     "\n",