diff --git a/nb_jax/environment.yml b/nb_jax/environment.yml
new file mode 100644
index 0000000000000000000000000000000000000000..f922592c609bd592680ae7804f79d473a58b6b4a
--- /dev/null
+++ b/nb_jax/environment.yml
@@ -0,0 +1,8 @@
+name: jax
+channels:
+  - conda-forge
+dependencies:
+  - numpy
+  - jupyterlab
+  - jax
+  - matplotlib