diff --git a/pyproject.toml b/pyproject.toml
index f50e056..0b88424 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -54,9 +54,16 @@ dev = [
     "tox>=4",
     "twine",
 ]
-autodiff = [
+autograd = [
     "autograd>=1.3",
-    "multiprocess>=0.7.0"
+    "multiprocess>=0.7.0",
+]
+jax = [
+    "jax>=0.4.1",
+    "multiprocess>=0.7.0",
+]
+symnum = [
+    "symnum>=0.2.0",
 ]
 
 [tool.coverage]