diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index 771c02cb87d8..ca6562db40ca 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -29,14 +29,14 @@
 
 buildPythonPackage rec {
   pname = "ax-platform";
-  version = "1.0.0";
+  version = "1.1.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "facebook";
     repo = "ax";
     tag = version;
-    hash = "sha256-DFsV1w6J7bTZNUq9OYExDvfc7IfTcthGKAnRMNujRKI=";
+    hash = "sha256-DHB/BcD913gano5N7dlAbB+Tfg1dMTRP3AR1HwJjkLg=";
   };
 
   env.ALLOW_BOTORCH_LATEST = "1";
@@ -57,11 +57,11 @@ buildPythonPackage rec {
     scikit-learn
     scipy
     sympy
-  ];
+  ]
+  ++ botorch.optional-dependencies.pymoo;
 
   nativeCheckInputs = [
     pyfakefs
-    # pytest-xdist
     pytestCheckHook
     sqlalchemy
     tabulate
@@ -74,7 +74,7 @@ buildPythonPackage rec {
     # broken with sqlalchemy 2
     "ax/core/tests/test_experiment.py"
     "ax/service/tests/test_ax_client.py"
-    "ax/service/tests/test_scheduler.py"
+    "ax/service/tests/test_orchestrator.py"
     "ax/service/tests/test_with_db_settings_base.py"
 
     # Hangs forever
diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix
index 4253117e7563..f2b60e3b4a26 100644
--- a/pkgs/development/python-modules/botorch/default.nix
+++ b/pkgs/development/python-modules/botorch/default.nix
@@ -3,15 +3,24 @@
   stdenv,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
+  setuptools,
+  setuptools-scm,
+
+  # dependencies
   gpytorch,
   linear-operator,
   multipledispatch,
   pyre-extensions,
   pyro-ppl,
-  setuptools,
-  setuptools-scm,
-  torch,
   scipy,
+  torch,
+
+  # optional-dependencies
+  pymoo,
+
+  # tests
   pytestCheckHook,
   pythonAtLeast,
 }:
@@ -43,6 +52,12 @@ buildPythonPackage rec {
     torch
   ];
 
+  optional-dependencies = {
+    pymoo = [
+      pymoo
+    ];
+  };
+
   nativeCheckInputs = [
     pytestCheckHook
   ];