skillfed

azureml-train-automl

Used for automatically finding the best machine learning model and its parameters.

azureml-train-automl v1.62.0 314.4K downloads/30d#7,700 on PyPI
License unclear https://aka.ms/azureml-sdk-license Active released

What it is and what it does

azureml-train-automl is Microsoft's automated machine learning package that removes the manual work of model selection and hyperparameter tuning. Given training and test data, it evaluates multiple algorithms and parameter combinations to identify the best-performing model for your task. The package is part of the Azure Machine Learning SDK and integrates with Azure's ML infrastructure through six runtime and core dependencies.

It targets developers and researchers who want to quickly establish a baseline model or explore the model space without hand-tuning each candidate. The package supports Python 3.8 through 3.11 on Windows, macOS, and Linux. Installation is straightforward (low friction), and the package is actively maintained in production-stable status.

Use it for:

  • Rapid prototyping: quickly find a working model for a new dataset without manual algorithm experimentation.
  • Baseline establishment: generate a strong baseline model to compare against custom or hand-tuned approaches.
  • Classification or regression: discover the best algorithm and parameters for tabular data tasks.
  • Production model selection: identify candidate models for deployment when integrated with Azure ML pipelines.
  • Exploratory model comparison: evaluate multiple algorithms automatically to understand which performs best.

Worth the install?

AI-flagged interpretation of the facts on this page — verify before relying

Automatically discovers and trains the best machine learning model and hyperparameters for your dataset, handling model selection and tuning without manual experimentation.

Yes, if you are already using Azure Machine Learning or plan to adopt it. The package delivers genuine value for automated model discovery and is actively maintained. No, if you need a standalone, cloud-agnostic AutoML solution—this package is tightly coupled to Azure infrastructure and requires proprietary licensing. Verify the license terms and Azure dependency requirements before committing.

Install

azureml-train-automl on PyPI

pip

pip install azureml-train-automl

uv

uv add azureml-train-automl

poetry

poetry add azureml-train-automl

Installing azureml-train-automl

Before you install

Low install friction with a pure-Python wheel. Actively maintained as of 170 days ago. Depends on six specialized azureml packages, which may require Azure SDK infrastructure already in place.

License in practice

Licensed under a proprietary Microsoft license (https://aka.ms/azureml-sdk-license). Verify terms before use in commercial or redistributed contexts.

Quickstart

pip install azureml-train-automl==1.62.0
from azureml.train.automl import AutoMLConfig

Requires Python 3.8 through 3.11 (capped below 3.12). Azure ML infrastructure and credentials may be required by runtime dependencies.

Verify before relying

  • Whether azureml-automl-core and related runtime dependencies are freely available or require Azure subscription.
  • Whether the package works offline or requires cloud connectivity to Azure services.
  • Performance characteristics and scalability limits for large datasets.

Package facts

License https://aka.ms/azureml-sdk-license (unclear)
Python support capped below the current Python release (<3.12,>=3.8)
Install friction low — pure-Python wheel
Runtime dependencies 6 — azureml-automl-core, azureml-automl-runtime, azureml-dataset-runtime, azureml-train-automl-client, azureml-train-automl-runtime, azureml-responsibleai
Maintenance actively maintained — 170 days since the last release
First released
Downloads 314,384/month — #7,700 on PyPI (30-day window, as of 2026-08-14)
Known vulnerabilities none known (OSV.dev, checked 2026-08-14)

Evidence: azureml_train_automl-1.62.0-py3-none-any.whl

Development Status :: 5 - Production/StableIntended Audience :: DevelopersIntended Audience :: Science/ResearchLicense :: Other/Proprietary LicenseOperating System :: MacOSOperating System :: Microsoft :: WindowsOperating System :: POSIX :: LinuxProgramming Language :: Python :: 3.10Programming Language :: Python :: 3.11Programming Language :: Python :: 3.8Programming Language :: Python :: 3.9Topic :: Scientific/Engineering :: Artificial Intelligence

Tags

automated machine learningautoml model selectionhyperparameter tuningautomatic model trainingml model optimizationbest model finderazure automl
automlazure-sdkhyperparameter-tuning

More Artificial Intelligence packages

litellm

LiteLLM provides a unified Python interface to…

permissive · top 100 on PyPI

huggingface-hub

Client library and CLI tool for downloading,…

permissive · top 100 on PyPI

langchain

LangChain provides a framework for building…

permissive · top 1,000 on PyPI

hf-xet

hf-xet provides chunk-based deduplication and…

permissive · top 1,000 on PyPI

tokenizers

Tokenizers converts raw text into token…

permissive · top 1,000 on PyPI

transformers

Transformers provides a unified framework for…

permissive · top 1,000 on PyPI

azureml-automl-core

Provides internal AutoML infrastructure and…

unclear · top 15,000 on PyPI

azureml-train-automl-client

Automatically selects and tunes machine…

unclear · top 15,000 on PyPI

FLAML

FLAML automates machine learning workflows by…

permissive · top 15,000 on PyPI

databricks-automl-runtime

Provides runtime support for Databricks AutoML,…

permissive · top 15,000 on PyPI

azureml-train-core

Provides estimator classes for training machine…

unclear · top 15,000 on PyPI

azureml-train-restclients-hyperdrive

Provides REST client classes for creating and…

unclear · top 15,000 on PyPI

azureml-sdk

Provides a Python SDK to build, train, and…

unclear · top 15,000 on PyPI

cloudml-hypertune

Reports hyperparameter tuning metrics to Google…

permissive · top 15,000 on PyPI

azureml-train

A deprecated wrapper package that redirects…

unclear · top 15,000 on PyPI

sweeps

Generates hyperparameter sweep suggestions…

permissive · top 15,000 on PyPI

Further reading