Shadowgraph
TrendingBuzzResearch
Chrome extension
Sign in

microsoft/nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Python

14,368

2,835 commits

updated Jul 3, 2024

nni.readthedocs.io

See the code

README


MIT licensed Issues Bugs Pull Requests Version Documentation Status

NNI automates feature engineering, neural architecture search, hyperparameter tuning, and model compression for deep learning. Find the latest features, API, examples and tutorials in our official documentation (简体中文版点这里).

What's NEW!  

Installation

See the NNI installation guide to install from pip, or build from source.

To install the current release:

$ pip install nni

To update NNI to the latest version, add --upgrade flag to the above commands.

NNI capabilities in a glance

Hyperparameter Tuning Neural Architecture Search Model Compression
Algorithms
Supported Frameworks Training Services Tutorials
Supports
  • PyTorch
  • TensorFlow
  • Scikit-learn
  • XGBoost
  • LightGBM
  • MXNet
  • Caffe2
  • More...
webui

Resources

Contribution guidelines

If you want to contribute to NNI, be sure to review the contribution guidelines, which includes instructions of submitting feedbacks, best coding practices, and code of conduct.

We use GitHub issues to track tracking requests and bugs. Please use NNI Discussion for general questions and new ideas. For questions of specific use cases, please go to Stack Overflow.

Participating discussions via the following IM groups is also welcomed.

GitterWeChat
imageORimage

Over the past few years, NNI has received thousands of feedbacks on GitHub issues, and pull requests from hundreds of contributors. We appreciate all contributions from community to make NNI thrive.

Test status

Essentials

TypeStatus
Fast testBuild Status
Full test - HPOBuild Status
Full test - NASBuild Status
Full test - compressionBuild Status

Training services

TypeStatus
Local - linuxBuild Status
Local - windowsBuild Status
Remote - linux to linuxBuild Status
Remote - windows to windowsBuild Status
OpenPAIBuild Status
FrameworkcontrollerBuild Status
KubeflowBuild Status
HybridBuild Status
AzureMLBuild Status

Targeting at openness and advancing state-of-art technology, Microsoft Research (MSR) had also released few other open source projects.

  • OpenPAI : an open source platform that provides complete AI model training and resource management capabilities, it is easy to extend and supports on-premise, cloud and hybrid environments in various scale.
  • FrameworkController : an open source general-purpose Kubernetes Pod Controller that orchestrate all kinds of applications on Kubernetes by a single controller.
  • MMdnn : A comprehensive, cross-framework solution to convert, visualize and diagnose deep neural network models. The "MM" in MMdnn stands for model management and "dnn" is an acronym for deep neural network.
  • SPTAG : Space Partition Tree And Graph (SPTAG) is an open source library for large scale vector approximate nearest neighbor search scenario.
  • nn-Meter : An accurate inference latency predictor for DNN models on diverse edge devices.

We encourage researchers and students leverage these projects to accelerate the AI development and research.

License

The entire codebase is under MIT license.

automated-machine-learning
automl
bayesian-optimization
data-science
deep-learning
deep-neural-network
distributed
feature-engineering
hyperparameter-optimization
hyperparameter-tuning
machine-learning
machine-learning-algorithms
mlops
model-compression
nas
neural-architecture-search
neural-network
python
pytorch
tensorflow

Project health

Overall

6.8/10

Maintained

10/10

Code review

10/10
OpenSSF Scorecard

PyPI

nni

Latest release

3.0.0 · Sep 14, 2023

Releases this year

0

Significant stargazers

(top 24 of 43)

David Sherret

2,130 followers · starred Mar 2019

Lukas Geiger

304 followers · starred Nov 2020

Pengfei Ni

2,781 followers · starred Sep 2018

Hamel Husain

2,861 followers · starred Feb 2019

Similar repos

microsoft/FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

4,407

same org

microsoft/qlib

Qlib is an AI-oriented Quant investment platform that aims to use AI tech to empower Quant…

49,313

same org

windmaple/awesome-AutoML

Curating a list of AutoML-related research, tools, projects and other resources

945

hibayesian/awesome-automl-papers

A curated list of automated machine learning papers, articles, tutorials, slides and projects

4,163

EthicalML/awesome-production-machine-learning

A curated list of awesome open source libraries to deploy, monitor, version and scale your machine…

21,000

openvinotoolkit/nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

1,208

1nslyn/autoMIL

Autonomous Agent-Driven Multiple Instance Learning

5

orgTestCodacy11KRepos110MB/repo-9238-katib

0

microsoft/nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

Python

14,368

2,835 commits

updated Jul 3, 2024

nni.readthedocs.io

See the code

README


MIT licensed Issues Bugs Pull Requests Version Documentation Status

NNI automates feature engineering, neural architecture search, hyperparameter tuning, and model compression for deep learning. Find the latest features, API, examples and tutorials in our official documentation (简体中文版点这里).

What's NEW!  

Installation

See the NNI installation guide to install from pip, or build from source.

To install the current release:

$ pip install nni

To update NNI to the latest version, add --upgrade flag to the above commands.

NNI capabilities in a glance

Hyperparameter Tuning Neural Architecture Search Model Compression
Algorithms
Supported Frameworks Training Services Tutorials
Supports
  • PyTorch
  • TensorFlow
  • Scikit-learn
  • XGBoost
  • LightGBM
  • MXNet
  • Caffe2
  • More...
webui

Resources

Contribution guidelines

If you want to contribute to NNI, be sure to review the contribution guidelines, which includes instructions of submitting feedbacks, best coding practices, and code of conduct.

We use GitHub issues to track tracking requests and bugs. Please use NNI Discussion for general questions and new ideas. For questions of specific use cases, please go to Stack Overflow.

Participating discussions via the following IM groups is also welcomed.

GitterWeChat
imageORimage

Over the past few years, NNI has received thousands of feedbacks on GitHub issues, and pull requests from hundreds of contributors. We appreciate all contributions from community to make NNI thrive.

Test status

Essentials

TypeStatus
Fast testBuild Status
Full test - HPOBuild Status
Full test - NASBuild Status
Full test - compressionBuild Status

Training services

TypeStatus
Local - linuxBuild Status
Local - windowsBuild Status
Remote - linux to linuxBuild Status
Remote - windows to windowsBuild Status
OpenPAIBuild Status
FrameworkcontrollerBuild Status
KubeflowBuild Status
HybridBuild Status
AzureMLBuild Status

Targeting at openness and advancing state-of-art technology, Microsoft Research (MSR) had also released few other open source projects.

  • OpenPAI : an open source platform that provides complete AI model training and resource management capabilities, it is easy to extend and supports on-premise, cloud and hybrid environments in various scale.
  • FrameworkController : an open source general-purpose Kubernetes Pod Controller that orchestrate all kinds of applications on Kubernetes by a single controller.
  • MMdnn : A comprehensive, cross-framework solution to convert, visualize and diagnose deep neural network models. The "MM" in MMdnn stands for model management and "dnn" is an acronym for deep neural network.
  • SPTAG : Space Partition Tree And Graph (SPTAG) is an open source library for large scale vector approximate nearest neighbor search scenario.
  • nn-Meter : An accurate inference latency predictor for DNN models on diverse edge devices.

We encourage researchers and students leverage these projects to accelerate the AI development and research.

License

The entire codebase is under MIT license.

automated-machine-learning
automl
bayesian-optimization
data-science
deep-learning
deep-neural-network
distributed
feature-engineering
hyperparameter-optimization
hyperparameter-tuning
machine-learning
machine-learning-algorithms
mlops
model-compression
nas
neural-architecture-search
neural-network
python
pytorch
tensorflow

Project health

Overall

6.8/10

Maintained

10/10

Code review

10/10
OpenSSF Scorecard

PyPI

nni

Latest release

3.0.0 · Sep 14, 2023

Releases this year

0

Significant stargazers

(top 24 of 43)

David Sherret

2,130 followers · starred Mar 2019

Lukas Geiger

304 followers · starred Nov 2020

Pengfei Ni

2,781 followers · starred Sep 2018

Hamel Husain

2,861 followers · starred Feb 2019

Similar repos

microsoft/FLAML

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

4,407

same org

microsoft/qlib

Qlib is an AI-oriented Quant investment platform that aims to use AI tech to empower Quant…

49,313

same org

windmaple/awesome-AutoML

Curating a list of AutoML-related research, tools, projects and other resources

945

hibayesian/awesome-automl-papers

A curated list of automated machine learning papers, articles, tutorials, slides and projects

4,163

EthicalML/awesome-production-machine-learning

A curated list of awesome open source libraries to deploy, monitor, version and scale your machine…

21,000

openvinotoolkit/nncf

Neural Network Compression Framework for enhanced OpenVINO™ inference

1,208

1nslyn/autoMIL

Autonomous Agent-Driven Multiple Instance Learning

5

orgTestCodacy11KRepos110MB/repo-9238-katib

0