From 558379c08e506b63ac9b5afcc7ade6925562a872 Mon Sep 17 00:00:00 2001
From: Steven Braun <steven.braun.mz@gmail.com>
Date: Sun, 5 Nov 2023 14:58:15 +0100
Subject: [PATCH] Update readme on notebooks

---
 README.md | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 66aa69a..3d6e1ca 100644
--- a/README.md
+++ b/README.md
@@ -4,7 +4,14 @@ This repository contains code for my personal, simplistic, EinsumNetworks implem
 
 For a speed benchmark comparison against the official EinsumNetworks implementation, check out [benchmark.md](./benchmark/benchmark.md) (short: simple-einet is faster in all dimensions except the input-channel size but scales similar to EinsumNetworks).
 
-# PyTorch Lightning Training
+## Notebooks
+
+The `notebooks` folder contains some Jupyter notebooks that demonstrate the usage of this library.
+
+- [Training an Einet on MNIST](./notebooks/mnist.ipynb)
+- [Training an Einet on synthetic multivariate Normal data](./notebooks/multivariate_normal.ipynb)
+
+## PyTorch Lightning Training
 
 The `main_pl.py` script offers PyTorch-Lightning based training for discriminative and generative Einets.