diff --git a/flake.lock b/flake.lock
new file mode 100644
index 0000000..c759433
--- /dev/null
+++ b/flake.lock
@@ -0,0 +1,61 @@
+{
+  "nodes": {
+    "flake-utils": {
+      "inputs": {
+        "systems": "systems"
+      },
+      "locked": {
+        "lastModified": 1731533236,
+        "narHash": "sha256-l0KFg5HjrsfsO/JpG+r7fRrqm12kzFHyUHqHCVpMMbI=",
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "rev": "11707dc2f618dd54ca8739b309ec4fc024de578b",
+        "type": "github"
+      },
+      "original": {
+        "owner": "numtide",
+        "repo": "flake-utils",
+        "type": "github"
+      }
+    },
+    "nixpkgs": {
+      "locked": {
+        "lastModified": 1770197578,
+        "narHash": "sha256-AYqlWrX09+HvGs8zM6ebZ1pwUqjkfpnv8mewYwAo+iM=",
+        "owner": "NixOS",
+        "repo": "nixpkgs",
+        "rev": "00c21e4c93d963c50d4c0c89bfa84ed6e0694df2",
+        "type": "github"
+      },
+      "original": {
+        "owner": "NixOS",
+        "ref": "nixos-unstable",
+        "repo": "nixpkgs",
+        "type": "github"
+      }
+    },
+    "root": {
+      "inputs": {
+        "flake-utils": "flake-utils",
+        "nixpkgs": "nixpkgs"
+      }
+    },
+    "systems": {
+      "locked": {
+        "lastModified": 1681028828,
+        "narHash": "sha256-Vy1rq5AaRuLzOxct8nz4T6wlgyUR7zLU309k9mBC768=",
+        "owner": "nix-systems",
+        "repo": "default",
+        "rev": "da67096a3b9bf56a91d16901293e51ba5b49a27e",
+        "type": "github"
+      },
+      "original": {
+        "owner": "nix-systems",
+        "repo": "default",
+        "type": "github"
+      }
+    }
+  },
+  "root": "root",
+  "version": 7
+}
diff --git a/flake.nix b/flake.nix
new file mode 100644
index 0000000..e12b827
--- /dev/null
+++ b/flake.nix
@@ -0,0 +1,54 @@
+{
+  description = "LakeBench dev environment";
+
+  inputs = {
+    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
+    flake-utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs = { self, nixpkgs, flake-utils }:
+    flake-utils.lib.eachDefaultSystem (system:
+      let
+        pkgs = import nixpkgs { inherit system; };
+        python = pkgs.python313;
+      in
+      {
+        devShells.default = pkgs.mkShell {
+          packages = [
+            python
+            pkgs.uv
+            pkgs.ruff
+          ];
+
+          shellHook = ''
+            create-venv() {
+              rm -rf .venv
+              uv venv --python ${python}/bin/python .venv
+              source .venv/bin/activate
+
+              uv pip install -e .
+              # spark and sail conflict with each other, install separately
+              for extra in duckdb polars daft tpcds_datagen tpch_datagen sparkmeasure spark sail; do
+                uv pip install "lakebench[$extra]" 2>&1 || echo "warning: $extra failed to install, skipping"
+              done
+              uv pip install --group dev
+              uv pip install jupyter ipykernel
+              python -m ipykernel install --user --name python3 --display-name "Python 3 (LakeBench)"
+
+              # store hash to detect pyproject.toml changes
+              md5sum pyproject.toml > .venv/.pyproject.hash
+            }
+
+            if [ ! -d .venv ]; then
+              create-venv
+            elif ! md5sum --check .venv/.pyproject.hash &>/dev/null; then
+              echo "pyproject.toml changed, recreating venv..."
+              create-venv
+            else
+              source .venv/bin/activate
+            fi
+          '';
+        };
+      }
+    );
+}