nix: Add development shell

flake.nix

@@ -0,0 +1,25 @@
+{
+  description = "SyntaxDot sequence labeler";
+
+  inputs = {
+    nixpkgs.url = "github:NixOS/nixpkgs/nixpkgs-unstable";
+    utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs = { self, nixpkgs, utils }:
+    utils.lib.eachSystem  [ "x86_64-linux" ] (system:
+    let
+      pkgs = import nixpkgs {
+        inherit system;
+        config = {
+          allowUnfreePredicate = pkg: builtins.elem (pkgs.lib.getName pkg) [
+            "libtorch"
+            "nvidia-x11"
+          ];
+          cudaSupport = true;
+        };
+      };
+    in {
+      devShell = import ./shell.nix { inherit pkgs; };
+    });
+}

shell.nix

@@ -0,0 +1,19 @@
+{ pkgs ? import <nixpkgs> {} }:
+
+with pkgs;
+
+mkShell {
+  nativeBuildInputs = [ cmake pkg-config rustup ];
+
+  buildInputs = [ openssl ];
+
+  HDF5_DIR = symlinkJoin {
+    name = "hdf5-join";
+    paths = [ hdf5.dev hdf5.out ];
+  };
+
+  LIBTORCH = symlinkJoin {
+    name = "torch-join";
+    paths = [ libtorch-bin.dev libtorch-bin.out ];
+  };
+}