diff --git a/spinoffs/autobnn/setup.py b/spinoffs/autobnn/setup.py
index 184fb98611..9e30e5126d 100644
--- a/spinoffs/autobnn/setup.py
+++ b/spinoffs/autobnn/setup.py
@@ -46,10 +46,13 @@
         'bayeux-ml',
         'chex',
         'flax',
+        'jax',
         'jaxtyping',
         'matplotlib',
+        'numpy',
         'pandas',
         'scipy',
+        'tensorflow-probability'
         ],
     # Add in any packaged data.
     exclude_package_data={'': ['BUILD']},