diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..f4a8c59
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,18 @@
+FROM docker.io/python:3.12.7-bookworm
+
+WORKDIR /workspace
+
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN pip install --no-cache-dir \
+    jax[cuda12] \
+    equinox \
+    torch \
+    optax \
+    torchvision
+
+CMD ["/usr/bin/bash"]