Download
Neural Network Intelligence is an open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning. NNI (Neural Network Intelligence) is a lightweight but powerful toolkit to help users automate feature engineering, neural architecture search, hyperparameter tuning and model compression. The tool manages automated machine learning (AutoML) experiments, dispatches and runs experiments' trial jobs generated by tuning algorithms to search the best neural architecture and/or hyper-parameters in different training environments like Local Machine, Remote Servers, OpenPAI, Kubeflow, FrameworkController on K8S (AKS etc.) DLWorkspace (aka. DLTS) AML (Azure Machine Learning) and other cloud options. NNI provides CommandLine Tool as well as an user friendly WebUI to manage training experiements.
Features
- Try different AutoML algorithms in their training code/model
- Run AutoML trial jobs in different environments to speed up search
- Implement and experiement new AutoML algorithms
- Implement hyperparameter tuning algorithms, neural architect search algorithms or model compression algorithms
- Customize your own AutoML algorithms and training services
- NNI also provides a set of build-in stat-of-the-art AutoML algorithms and out of box support for popular training platforms
Project Samples
