diff --git a/environment.lock.yml b/environment.lock.yml
index 277fe961..e52ef921 100644
--- a/environment.lock.yml
+++ b/environment.lock.yml
@@ -1,6 +1,7 @@
 name: cuda-downscaling
 channels:
   - pytorch
+  - pyviz
   - conda-forge
   - defaults
 dependencies:
@@ -364,6 +365,7 @@ dependencies:
   - pip:
       - atpublic==3.1.1
       - flufl-lock==7.1.1
+      - mlde-utils==0.0.4
       - netcdf4==1.6.3
       - python-cmethods==1.0.1
 prefix: /home/henry/mambaforge/envs/cuda-downscaling
diff --git a/pyproject.toml b/pyproject.toml
index baa99dd8..36497d76 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -16,10 +16,7 @@ classifiers = [
     "License :: OSI Approved :: Apache Software License",
     "Operating System :: OS Independent",
 ]
-dependencies = [
-  "flufl-lock",
-  "python-cmethods"
-]
+dynamic = ["dependencies"]
 
 [project.urls]
 "Homepage" = "https://github.com/henryaddison/score_sde_pytorch"
@@ -28,6 +25,8 @@ dependencies = [
 [project.scripts]
     mlde = "ml_downscaling_emulator.bin:app"
 
+[tool.setuptools.dynamic]
+dependencies = { file = ["requirements.txt"] }
 
 [tool.black]
 extend-exclude = '^/src/ml_downscaling_emulator/score_sde_pytorch_hja22/'
diff --git a/requirements.txt b/requirements.txt
index f93b8bef..afe7906f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,11 +1,3 @@
-ml-collections==0.1.0
-tensorflow-gan==2.0.0
-tensorflow_io
-tensorflow_datasets==3.1.0
-tensorflow==2.4.0
-tensorflow-addons==0.12.0
-tensorboard==2.4.0
-absl-py==0.10.0
-torch>=1.7.0
-torchvision
-ninja
+flufl-lock
+mlde-utils ~=0.0.4
+python-cmethods