msranni/nni repository overview

⁠What is NNI?

NNI (Neural Network Intelligence) is a toolkit to help users run automated machine learning (AutoML) experiments. The tool dispatches and runs trial jobs generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different environments like local machine, remote servers and cloud.

⁠Who should consider using NNI

• Those who want to try different AutoML algorithms in their training code (model) at their local machine.
• Those who want to run AutoML trial jobs in different environments to speed up search (e.g. remote servers and cloud).
• Researchers and data scientists who want to implement their own AutoML algorithms and compare it with other algorithms.
• ML Platform owners who want to support AutoML in their platform.

⁠Documentation

• NNI overview⁠
• Quick start⁠

⁠Github

• Homepage⁠
• Wiki⁠

⁠Supported Libraries

Ubuntu 16.04 LTS CUDA 9.0, CuDNN 7.0 numpy 1.14.3,scipy 1.1.0 TensorFlow 1.5.0 Keras 2.1.6 PyTorch 0.4.1 scikit-learn 0.2.0 NNI v0.4.1

⁠References

https://github.com/Microsoft/nni/tree/master/deployment/docker⁠