diff --git a/pkgs/development/python-modules/ax/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
similarity index 93%
rename from pkgs/development/python-modules/ax/default.nix
rename to pkgs/development/python-modules/ax-platform/default.nix
index 7b3cc8f2b1c2..079dcdaf3ebf 100644
--- a/pkgs/development/python-modules/ax/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -2,7 +2,7 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
-  ax,
+  ax-platform,
   botorch,
   ipywidgets,
   jinja2,
@@ -22,13 +22,13 @@
 }:
 
 buildPythonPackage rec {
-  pname = "ax";
+  pname = "ax-platform";
   version = "0.4.0";
   format = "pyproject";
 
   src = fetchFromGitHub {
     owner = "facebook";
-    repo = pname;
+    repo = "ax";
     rev = "refs/tags/${version}";
     hash = "sha256-dj6Gig8N4oLtcZLwPl4QDHG/FwA2nFBtYxSARnWiJJU=";
   };
@@ -86,7 +86,7 @@ buildPythonPackage rec {
   # Many portions of the test suite fail under Python 3.12
   doCheck = lib.versions.majorMinor python.version != "3.12";
 
-  passthru.tests.check = ax.overridePythonAttrs { doCheck = true; };
+  passthru.tests.check = ax-platform.overridePythonAttrs { doCheck = true; };
 
   meta = with lib; {
     description = "Ax is an accessible, general-purpose platform for understanding, managing, deploying, and automating adaptive experiments";
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index b515694b0b91..db7cac08f5b5 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -1088,7 +1088,7 @@ self: super: with self; {
 
   awswrangler = callPackage ../development/python-modules/awswrangler { };
 
-  ax = callPackage ../development/python-modules/ax { };
+  ax-platform = callPackage ../development/python-modules/ax-platform { };
 
   axis = callPackage ../development/python-modules/axis { };