diff --git a/pyproject.toml b/pyproject.toml
index 552a91f..e4d1b32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,6 +29,7 @@ dependencies = [
   "omegaconf>=2.3.0",
   "pyyaml>=6.0.2",
   "hydra-core>=1.3.2",
+  "numba>=0.60",
 ]
 requires-python = ">=3.9,<3.13"
 readme = "README.md"