NNI

Overview

Open source AutoML toolkit that automates machine learning workflows including feature engineering, neural architecture search, model compression, and hyperparameter tuning. Written in Python and maintained by the community. Handles the repetitive optimization tasks in the ML lifecycle to reduce manual experimentation.

Best for

Best for
ML engineers and researchers who need flexible, self-hosted AutoML for hyperparameter tuning and neural architecture search

Use cases

• Hyperparameter tuning at scale across distributed systems
• Neural architecture search for deep learning models
• Model compression and optimization for deployment

Notes

Open source AutoML toolkit that automates machine learning workflows including feature engineering, neural architecture search, model compression, and hyperparameter tuning. Written in Python and maintained by the community. Handles the repetitive optimization tasks in the ML lifecycle to reduce manual experimentation.

14,352 stars on GitHub. Last updated 2024-07-03. Licensed MIT.

Use cases

• Hyperparameter tuning at scale across distributed systems
• Neural architecture search for deep learning models
• Model compression and optimization for deployment

Pros

• Supports distributed tuning across multiple machines and GPUs
• Covers full ML lifecycle from feature engineering to model compression
• Active open source project with 14k+ GitHub stars

Cons

• Requires Python expertise and familiarity with ML concepts to configure effectively
• Community-maintained with no commercial support guarantees
• Steeper learning curve compared to managed AutoML services

Indexed from awesome-llmops and enriched against its public facts.