diff --git a/pyproject.toml b/pyproject.toml
index 94481e1..1c92882 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "toto-ts"
-version = "0.1.4"
+version = "0.2.0"
 description = "Time-Series-Optimized Transformer for Observability"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -12,10 +12,43 @@ license = {text = "Apache-2.0"}
 authors = [
     {name = "DataDog, Inc."}
 ]
-dynamic = ["dependencies"]
+dependencies = [
+    "torch[cuda]>=2.7.0,<3",
+    "einops>=0.7.0,<1",
+    "gluonts[torch]>=0.15.1,<1",
+    "jaxtyping>=0.2.29,<1",
+    "pandas>=2.2.3,<3",
+    "rotary-embedding-torch~=0.8.6",
+    "types-tabulate==0.9.0.20241207",
+    "tqdm>=4.66.3,<5",
+    "tabulate>=0.9.0,<1",
+    "transformers~=4.52.1",
 
-[tool.setuptools.dynamic]
-dependencies = {file = ["requirements.txt"]}
+
+]
+
+[project.optional-dependencies]
+all = [
+    "aioboto3~=12.4.0",
+    "boto3~=1.34.69",
+    "datasets~=2.17.1",
+    "dill~=0.3.8",
+    "scikit-learn>=1.5.0,<2",
+    "PyYAML>=6.0.2,<7",
+    "types-PyYAML==6.0.12.20240917",
+
+]
+dev = [
+    "beartype~=0.18.5",
+    "mypy>=1.11.1,<2",
+    "pytest>=8.3.5,<9",
+    "pytest-env>=1.1.5,<2",
+    "mypy>=1.11.1,<2",
+    "black~=24.10.0",
+    "isort~=5.13.2",
+    "jupyter~=1.1.1",
+    "matplotlib>=3.9.2,<4",
+]
 
 [tool.setuptools.packages.find]
 include = ["*"]
diff --git a/requirements.txt b/requirements.txt
deleted file mode 100644
index 58e29be..0000000
--- a/requirements.txt
+++ /dev/null
@@ -1,28 +0,0 @@
-torch==2.7.0
-aioboto3==12.4.0
-beartype==0.18.5
-boto3==1.34.69
-datasets==2.17.1
-dill==0.3.8
-einops==0.7.0
-gluonts[torch]==0.15.1
-jaxtyping==0.2.29
-mypy==1.11.1
-pandas==2.2.3
-pytest==8.3.5
-pytest-env==1.1.5
-rotary-embedding-torch==0.8.6
-scikit-learn==1.5.0
-types-tabulate==0.9.0.20241207
-transformers==4.52.1
-tqdm==4.66.3
-beartype==0.18.5
-mypy==1.11.1
-pyyaml==6.0.2
-types-PyYAML==6.0.12.20240917
-black==24.10.0
-isort==5.13.2
-PyYAML==6.0.2
-jupyter==1.1.1
-matplotlib==3.9.2
-tabulate==0.9.0