diff --git a/pyproject.toml b/pyproject.toml
index 5b07ad9..6eee5f2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,6 +20,12 @@ accelerate = "^0.25.0"
 sentencepiece = "^0.1.99"
 scipy = "^1.11.4"
 
+[tool.poetry.group.dev.dependencies]
+pytest = "^7.4.3"
+black = "^23.12.1"
+pre-commit = "^3.6.0"
+jupyterlab = "^4.0.9"
+ipywidgets = "^8.1.1"
 
 [build-system]
 requires = ["poetry-core"]