Update dependency versions for PyMC v5

setup.py

@@ -23,23 +23,26 @@
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
 ]
-# TODO: Update exoplanet-core version for pymc5
-INSTALL_REQUIRES = ["exoplanet-core>=0.2.0", "astropy>=3.1"]
+INSTALL_REQUIRES = ["astropy>=3.1"]
 EXTRA_REQUIRE = {
     "pymc3": [
         "pymc3>=3.9",
         "numpy<1.22",
         "aesara-theano-fallback>=0.0.2",
         "xarray<2023.10.0",
+        "exoplanet-core>=0.2.0",
+    ],
+    "pymc": ["pymc>=5.0.0", "exoplanet-core>=0.3.0"],
+    "extras": ["pymc>=5.0.0", "pymc-ext>=1.0.1", "celerite2>=0.3.1"],
+    "extras-pymc3": [
+        "pymc3>=3.9",
+        "pymc3-ext>=0.1.0",
+        "celerite2>=0.2.0,<0.3.1",
     ],
-    "pymc": ["pymc>=5.0.0"],
-    # TODO: Update pymc-ext and celerite2 versions for pymc5
-    "extras": ["pymc>=5.0.0", "pymc-ext>=1.0.0", "celerite2>=0.2.0"],
     "test": ["scipy", "pytest", "pytest-env"],
-    # TODO: Update pymc-ext version for pymc5
     "docs": [
         "pymc>=5.0.0",
-        "pymc-ext>=1.0.0",
+        "pymc-ext>=1.0.1",
         "sphinx-book-theme",
         "myst-nb",
         "ipython",