diff --git a/crispio/cli.py b/crispio/cli.py
index d5def17..71f5298 100644
--- a/crispio/cli.py
+++ b/crispio/cli.py
@@ -19,7 +19,7 @@
 from .map import GuideLibrary
 from .utils import sequences
 
-__version__ = '0.0.1'
+__version__ = '0.0.2'
 
 def _allow_broken_pipe(f: Callable) -> Callable:
 
diff --git a/pyproject.toml b/pyproject.toml
index 09c3f84..18cfd7d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "crispio"
-version = "0.0.1"
+version = "0.0.2"
 authors = [
   { name="Eachan Johnson", email="eachan.johnson@crick.ac.uk" },
 ]
@@ -29,7 +29,6 @@ classifiers = [
 
 dependencies = [ 
   "bioino>=0.0.2",
-  "carabiner-tools",
   "pyyaml",
   "nemony",
   "numpy",
@@ -41,7 +40,7 @@ dependencies = [
 [project.optional-dependencies]
 fit = [
   "carabiner-tools[pd,mpl]",
-  "jax",
+  "jax[cuda12]",
   "optax",
   "scipy",
   "matplotlib",