diff --git a/flake.nix b/flake.nix
index f6b0935..90c0262 100644
--- a/flake.nix
+++ b/flake.nix
@@ -39,6 +39,7 @@
             pydot
             graphviz
             (python311.pkgs.callPackage ./pkgs/azureml-mlflow/default.nix {})
+            (python311.pkgs.callPackage ./pkgs/azure-ai-ml/default.nix {})
           ]);
       };
     });
diff --git a/pkgs/azure-ai-ml/default.nix b/pkgs/azure-ai-ml/default.nix
new file mode 100644
index 0000000..e3de78e
--- /dev/null
+++ b/pkgs/azure-ai-ml/default.nix
@@ -0,0 +1,33 @@
+{
+  lib,
+  buildPythonPackage,
+  fetchPypi,
+  setuptools,
+  python311,
+}:
+buildPythonPackage rec {
+  pname = "azure_ai_ml";
+  version = "1.19.0";
+  format = "wheel";
+
+  src = fetchPypi {
+    inherit pname version format;
+    sha256 = "sha256-8Dha8G776uH4MRNhPkU0NQjRKI/S8FhXYZ58fU1PUwI=";
+    dist = "py3";
+    python = "py3";
+  };
+
+  nativeBuildInputs = [setuptools];
+
+  propagatedBuildInputs = [
+  ];
+
+  doCheck = false; # Package does not contain tests
+
+  meta = with lib; {
+    description = "The azureml-mlflow package contains the integration code of AzureML with MLflow. MLflow (https://mlflow.org/) is an open-source platform for tracking machine learning experiments and managing models. You can use MLflow logging APIs with Azure Machine Learning so that metrics and artifacts are logged to your Azure machine learning workspace.";
+    homepage = "https://docs.microsoft.com/python/api/overview/azure/ml/?view=azure-ml-py";
+    license = licenses.mit;
+    maintainers = with maintainers; [Lillian-Violet];
+  };
+}