Add wandb sweep helper (#133)

Author: alan-cooney

.vscode/cspell.json

@@ -42,6 +42,7 @@
         "intuniform",
         "invloguniform",
         "invloguniformvalues",
+        "ipynb",
         "itemwise",
         "jaxtyping",
         "kaiming",

docs/content/SUMMARY.md

@@ -1,6 +1,7 @@
 
 * [Home](index.md)
 * [Demo](demo.ipynb)
+* [Flexible Demo](flexible_demo.ipynb)
 * [Reference](reference/)
 * [Contributing](contributing.md)
 * [Citation](citation.md)