From ca4ad23a2b6b239a549380eb3800c03e05585238 Mon Sep 17 00:00:00 2001
From: vdeltatto <vdeltatt@sissa.it>
Date: Wed, 13 Nov 2024 13:31:21 +0100
Subject: [PATCH] add missing jax-DII docs for readthedocs

---
 docs/source/diff_imbalance.rst       | 5 +++++
 docs/source/jupyter_example_7.nblink | 4 ++++
 2 files changed, 9 insertions(+)
 create mode 100644 docs/source/diff_imbalance.rst
 create mode 100644 docs/source/jupyter_example_7.nblink

diff --git a/docs/source/diff_imbalance.rst b/docs/source/diff_imbalance.rst
new file mode 100644
index 00000000..4e8a01ce
--- /dev/null
+++ b/docs/source/diff_imbalance.rst
@@ -0,0 +1,5 @@
+The diff_imbalance module
+================================
+
+.. automodule:: diff_imbalance
+    :members:
\ No newline at end of file
diff --git a/docs/source/jupyter_example_7.nblink b/docs/source/jupyter_example_7.nblink
new file mode 100644
index 00000000..56929f98
--- /dev/null
+++ b/docs/source/jupyter_example_7.nblink
@@ -0,0 +1,4 @@
+{
+    "path": "../../examples/notebook_on_differentiable_imbalance_jax.ipynb"
+
+}
\ No newline at end of file