From 74b6843fdc0c0e5064fb3e53acf64cf12cb71210 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 2 Jul 2026 08:21:00 +0000
Subject: [PATCH 1/3] python3Packages.botorch: 0.17.2 -> 0.18.1

Diff: https://github.com/meta-pytorch/botorch/compare/v0.17.2...v0.18.1

Changelog: https://github.com/meta-pytorch/botorch/blob/v0.18.1/CHANGELOG.md
---
 pkgs/development/python-modules/botorch/default.nix | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix
index 827e86b7151a..1cc078b4224a 100644
--- a/pkgs/development/python-modules/botorch/default.nix
+++ b/pkgs/development/python-modules/botorch/default.nix
@@ -12,8 +12,8 @@
   gpytorch,
   linear-operator,
   multipledispatch,
+  ninja,
   pyre-extensions,
-  pyro-ppl,
   scipy,
   threadpoolctl,
   torch,
@@ -23,20 +23,23 @@
   pymoo,
 
   # tests
+  jax,
+  numpyro,
   pytestCheckHook,
   pythonAtLeast,
 }:
 
 buildPythonPackage (finalAttrs: {
   pname = "botorch";
-  version = "0.17.2";
+  version = "0.18.1";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "meta-pytorch";
     repo = "botorch";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-KWkRdgOAhWoVRoBn0XvKhTQX3KPHJrzqWoiaKrYzi7o=";
+    hash = "sha256-tUTtImqPlbS8g1oLoTTCCWQbeLwLop13qPjFrQeMtb8=";
   };
 
   build-system = [
@@ -48,8 +51,8 @@ buildPythonPackage (finalAttrs: {
     gpytorch
     linear-operator
     multipledispatch
+    ninja
     pyre-extensions
-    pyro-ppl
     scipy
     threadpoolctl
     torch
@@ -63,6 +66,8 @@ buildPythonPackage (finalAttrs: {
   };
 
   nativeCheckInputs = [
+    jax
+    numpyro
     pytestCheckHook
   ];
 

From d29f897746e2fa6c265626549f0df6a34ea3bb74 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 2 Jul 2026 08:12:42 +0000
Subject: [PATCH 2/3] python3Packages.ax-platform: 1.2.4 -> 1.3.1

Diff: https://github.com/facebook/ax/compare/1.2.4...1.3.1

Changelog: https://github.com/facebook/Ax/releases/tag/1.3.1
---
 pkgs/development/python-modules/ax-platform/default.nix | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index a82550a8de9c..aee136f5c4e8 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -22,6 +22,8 @@
   sympy,
 
   # tests
+  jax,
+  numpyro,
   pyfakefs,
   pytestCheckHook,
   sqlalchemy,
@@ -30,14 +32,15 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "ax-platform";
-  version = "1.2.4";
+  version = "1.3.1";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "facebook";
     repo = "ax";
     tag = finalAttrs.version;
-    hash = "sha256-ZeYoLOPF2D1bk89V5/WO/v0UCtDisbAOOL/Su9sB2Fg=";
+    hash = "sha256-8Q/p74rzMAciNUjaTJtf6duslc3TXT7MIjcHt2sffqI=";
   };
 
   env.ALLOW_BOTORCH_LATEST = "1";
@@ -63,6 +66,8 @@ buildPythonPackage (finalAttrs: {
   ++ botorch.optional-dependencies.pymoo;
 
   nativeCheckInputs = [
+    jax
+    numpyro
     pyfakefs
     pytestCheckHook
     sqlalchemy

From 178437a3bbc9e9ef21e486e552045f6a5f2dcd4b Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 2 Jul 2026 08:14:05 +0000
Subject: [PATCH 3/3] python3Packages.ax-platform: add missing test dependency

---
 pkgs/development/python-modules/ax-platform/default.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index aee136f5c4e8..6254cf13b416 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -28,6 +28,7 @@
   pytestCheckHook,
   sqlalchemy,
   tabulate,
+  tensorboard,
 }:
 
 buildPythonPackage (finalAttrs: {
@@ -72,6 +73,7 @@ buildPythonPackage (finalAttrs: {
     pytestCheckHook
     sqlalchemy
     tabulate
+    tensorboard
   ];
 
   disabledTestPaths = [