draft validation notebook

renecotyfanboy · May 11, 2024 · dc9b2d9 · dc9b2d9
1 parent 8569af5
commit dc9b2d9
Show file tree

Hide file tree

Showing 3 changed files with 23,136 additions and 4 deletions.
diff --git a/.gitignore b/.gitignore
@@ -10,7 +10,6 @@ main.py
 
 .ipynb_checkpoints
 */.ipynb_checkpoints/*
-*.ipynb
 poetry.lock
 # IPython
 profile_default/
@@ -197,4 +196,4 @@ cython_debug/
 .Trashes
 ehthumbs.db
 Thumbs.db
-# End of https://www.toptal.com/developers/gitignore/api/python,jupyternotebooks
+# End of https://www.toptal.com/developers/gitignore/api/python,jupyternotebooks
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,6 @@
 numpy 
 matplotlib
-jax
-jaxlib
+jax[cuda12]
 numpyro
 tqdm
 plotly