qutip
diff --git a/‎.pre-commit-config.yaml
+2 b/‎.pre-commit-config.yaml
+2
diff --git a/‎requirements.txt
+2 b/‎requirements.txt
+2
diff --git a/‎setup.cfg
+2 b/‎setup.cfg
+2
@@ -92,3 +92,5 @@ repos:
           - jaxlib
           - diffrax
           - pytest
+          - gymnasium
+          - stable-baselines3
@@ -7,3 +7,5 @@ qutip>=5.0.1
 qutip-qtrl
 qutip-jax
 pre-commit
+gymnasium>=0.29.1
+stable-baselines3>=2.3.2
@@ -35,6 +35,8 @@ install_requires =
     qutip-qtrl
     qutip-jax
     numpy>=1.16.6,<2.0
+    gymnasium>=0.29.1
+    stable-baselines3>=2.3.2
 setup_requires =
     cython>=1.0
     packaging