diff --git a/pyproject.toml b/pyproject.toml
index 4972f09ef..4ace9b7e8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -36,7 +36,6 @@ dependencies = [
    "jsonschema",
    "requests",
    "tqdm",
-   "datasets",
    "typing_extensions",
    "pycountry",
    "airportsdata",
@@ -46,6 +45,12 @@ dependencies = [
 dynamic = ["version"]
 
 [project.optional-dependencies]
+vllm = ["vllm"],
+transformers = ["transformers", "accelerate", "datasets"]
+mlxlm = ["mlx-lm", "datasets"]
+openai = ["openai"]
+llamacpp = ["llama-cpp-python", "transformers", "datasets"]
+exllamav2 = ["exllamav2"]
 test = [
     "pre-commit",
     "pytest",