microsoft/nni

Health v2 · maintenance only

Flags: archived

Adoption not part of the score

14361 stars · 1850 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

NNI (Neural Network Intelligence) is an open-source AutoML toolkit from Microsoft that automates hyperparameter tuning, neural architecture search, model compression (pruning and quantization), and feature engineering. It supports PyTorch and TensorFlow and can run experiments locally or distributed across clusters.

Use cases

• automate hyperparameter tuning for deep learning models
• run neural architecture search experiments
• prune and quantize models for faster inference
• compress a PyTorch model while preserving accuracy
• run distributed AutoML experiments across a cluster
• automate feature engineering for tabular data
• compare tuning algorithms like Bayesian optimization

When to choose

• you want a mature, feature-rich AutoML toolkit covering HPO, NAS, and model compression in one package
• you work in PyTorch or TensorFlow and want plug-and-play tuning with a web dashboard
• you need distributed hyperparameter search across multiple machines

When to avoid

• you need a lightweight single-purpose HPO tool like Optuna with minimal setup
• you require cutting-edge features or frequent updates, as development activity has slowed
• you work outside Python or with frameworks other than PyTorch/TensorFlow

Facets

library · maturity maintenance

machine-learning deep-learning benchmarking data-science machine-learning deep-learning data-science artificial-intelligence python windows automl hyperparameter-optimization neural-architecture-search model-compression pruning quantization feature-engineering pytorch tensorflow distributed-training linux macos gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="microsoft/nni")