azureml-train-automl
Used for automatically finding the best machine learning model and its parameters.
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-automluv
uv add azureml-train-automlpoetry
poetry add azureml-train-automlInstalling 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
Tags
More Artificial Intelligence packages
LiteLLM provides a unified Python interface to…
permissive · top 100 on PyPI
huggingface-hubClient library and CLI tool for downloading,…
permissive · top 100 on PyPI
langchainLangChain provides a framework for building…
permissive · top 1,000 on PyPI
hf-xethf-xet provides chunk-based deduplication and…
permissive · top 1,000 on PyPI
tokenizersTokenizers converts raw text into token…
permissive · top 1,000 on PyPI
transformersTransformers provides a unified framework for…
permissive · top 1,000 on PyPI
azureml-automl-coreProvides internal AutoML infrastructure and…
unclear · top 15,000 on PyPI
azureml-train-automl-clientAutomatically selects and tunes machine…
unclear · top 15,000 on PyPI
FLAMLFLAML automates machine learning workflows by…
permissive · top 15,000 on PyPI
databricks-automl-runtimeProvides runtime support for Databricks AutoML,…
permissive · top 15,000 on PyPI
azureml-train-coreProvides estimator classes for training machine…
unclear · top 15,000 on PyPI
azureml-train-restclients-hyperdriveProvides REST client classes for creating and…
unclear · top 15,000 on PyPI
azureml-sdkProvides a Python SDK to build, train, and…
unclear · top 15,000 on PyPI
cloudml-hypertuneReports hyperparameter tuning metrics to Google…
permissive · top 15,000 on PyPI
azureml-trainA deprecated wrapper package that redirects…
unclear · top 15,000 on PyPI
sweepsGenerates hyperparameter sweep suggestions…
permissive · top 15,000 on PyPI