microsoft__nni
Science Score: 44.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
✓CITATION.cff file
Found CITATION.cff file -
✓codemeta.json file
Found codemeta.json file -
✓.zenodo.json file
Found .zenodo.json file -
○DOI references
-
○Academic publication links
-
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (12.4%) to scientific vocabulary
Last synced: 9 months ago
·
JSON representation
·
Repository
Basic Info
- Host: GitHub
- Owner: swe-train
- License: mit
- Language: Python
- Default Branch: debug-ws
- Size: 127 MB
Statistics
- Stars: 0
- Watchers: 0
- Forks: 0
- Open Issues: 0
- Releases: 0
Created about 2 years ago
· Last pushed almost 2 years ago
Metadata Files
Readme
License
Citation
Security
README.md
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! 
- New release: v3.0 preview is available - released on May-5-2022
- New demo available: Youtube entry | Bilibili 入口 - last updated on June-22-2022
- New research paper: SparTA: Deep-Learning Model Sparsity via Tensor-with-Sparsity-Attribute - published in OSDI 2022
- New research paper: Privacy-preserving Online AutoML for Domain-Specific Face Detection - published in CVPR 2022
- Newly upgraded documentation: Doc upgraded
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 |
|
|
Resources
- NNI Documentation Homepage
- NNI Installation Guide
- NNI Examples
- Python API Reference
- Releases (Change Log)
- Related Research and Publications
- Youtube Channel of NNI
- Bilibili Space of NNI
- Webinar of Introducing Retiarii: A deep learning exploratory-training framework on NNI
- Community Discussions
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.
|Gitter||WeChat|
|----|----|----|
|
| OR |
|
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
| Type | Status |
| :---: | :---: |
| Fast test | |
| Full test - HPO |
|
| Full test - NAS |
|
| Full test - compression |
|
Training services
| Type | Status |
| :---: | :---: |
| Local - linux | |
| Local - windows |
|
| Remote - linux to linux |
|
| Remote - windows to windows |
|
| OpenPAI |
|
| Frameworkcontroller |
|
| Kubeflow |
|
| Hybrid |
|
| AzureML |
|
Related Projects
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.
Owner
- Name: swe-train
- Login: swe-train
- Kind: organization
- Repositories: 1
- Profile: https://github.com/swe-train
Citation (CITATION.cff)
cff-version: 1.2.0 message: "If you use this software, please cite it as below." authors: - name: "Microsoft" title: "Neural Network Intelligence" date-released: 2021-01-14 url: "https://github.com/microsoft/nni" version: 2.0
GitHub Events
Total
Last Year
Dependencies
Dockerfile
docker
- nvidia/cuda 11.3.1-cudnn8-runtime-ubuntu20.04 build
examples/trials/systems_auto_tuning/opevo/Dockerfile
docker
- nvidia/cuda 10.0-cudnn7-devel-ubuntu18.04 build
ts/jupyter_extension/package.json
npm
- @jupyterlab/builder 3.0.9 development
- @jupyterlab/application <3.1.0
- @jupyterlab/launcher <3.1.0
ts/jupyter_extension/yarn.lock
npm
- 433 dependencies
ts/nni_manager/package-lock.json
npm
- 837 dependencies
ts/nni_manager/package.json
npm
- @types/app-module-path >=2.2.0 development
- @types/chai >=4.2.18 development
- @types/chai-as-promised >=7.1.0 development
- @types/express >=4.17.14 development
- @types/express-ws >=3.0.1 development
- @types/glob >=8.0.0 development
- @types/http-proxy >=1.17.7 development
- @types/js-yaml >=4.0.1 development
- @types/lockfile >=1.0.0 development
- @types/mocha >=8.2.3 development
- @types/node >=18.15.0 development
- @types/request >=2.48.5 development
- @types/rx >=4.1.2 development
- @types/sqlite3 >=3.1.8 development
- @types/ssh2 >=0.5.52 development
- @types/stream-buffers >=3.0.3 development
- @types/tar >=6.1.3 development
- @types/tmp >=0.2.0 development
- @types/ws >=8.5.3 development
- @types/yargs >=17.0.13 development
- @typescript-eslint/eslint-plugin >=5.42.0 development
- @typescript-eslint/parser >=5.42.0 development
- chai >=4.3.6 development
- chai-as-promised >=7.1.1 development
- eslint >=8.26.0 development
- mocha >=10.1.0 development
- node-gyp >=9.3.0 development
- npm >=9.6.1 development
- nyc >=15.1.0 development
- request >=2.88.2 development
- tmp >=0.2.1 development
- ts-node >=10.9.1 development
- typescript >=4.8.4 development
- app-module-path ^2.2.0
- azure-storage ^2.10.7
- child-process-promise ^2.2.1
- express ^4.18.2
- express-joi-validator ^2.0.1
- express-ws ^5.0.2
- glob ^8.1.0
- http-proxy ^1.18.1
- ignore ^5.2.4
- js-base64 ^3.7.5
- js-yaml ^4.1.0
- kubernetes-client ^6.12.1
- lockfile ^1.0.4
- python-shell ^3.0.0
- rx ^4.1.0
- sqlite3 ^5.1.6
- ssh2 ^1.11.0
- stream-buffers ^3.0.2
- tail-stream ^0.3.4
- tar ^6.1.13
- tree-kill ^1.2.2
- ts-deferred ^1.0.4
- typescript-ioc ^1.2.6
- typescript-string-operations ^1.4.1
- ws ^8.13.0
- yargs ^17.7.1
ts/webui/package-lock.json
npm
- 1443 dependencies
ts/webui/package.json
npm
- @babel/core >=7.21.3 development
- @babel/eslint-parser >=7.21.3 development
- @babel/preset-react >=7.18.6 development
- @babel/preset-typescript >=7.21.0 development
- @babel/runtime >=7.21.0 development
- @svgr/webpack >=6.5.1 development
- @types/d3 >=7.4.0 development
- @types/node >=18.15.3 development
- @types/react >=17.0.53 development
- @types/react-dom >=17.0.19 development
- @typescript-eslint/eslint-plugin >=5.55.0 development
- @typescript-eslint/parser >=5.55.0 development
- babel-jest >=29.5.0 development
- babel-loader >=9.1.2 development
- babel-plugin-named-asset-import >=0.3.8 development
- babel-preset-react-app >=10.0.1 development
- case-sensitive-paths-webpack-plugin >=2.4.0 development
- concurrently >=7.6.0 development
- css-loader >=5.2.7 development
- css-minimizer-webpack-plugin >=4.2.2 development
- dotenv >=16.0.3 development
- dotenv-expand >=10.0.0 development
- eslint >=8.36.0 development
- eslint-config-prettier >=8.7.0 development
- eslint-config-react-app >=7.0.1 development
- eslint-plugin-flowtype >=8.0.3 development
- eslint-plugin-import >=2.27.5 development
- eslint-plugin-jsx-a11y >=6.7.1 development
- eslint-plugin-prettier >=4.2.1 development
- eslint-plugin-react >=7.32.2 development
- eslint-plugin-react-hooks >=4.6.0 development
- eslint-webpack-plugin >=4.0.0 development
- express >=4.18.2 development
- file-loader >=6.2.0 development
- fs-extra >=11.1.0 development
- html-webpack-plugin >=5.5.0 development
- jest >=29.5.0 development
- mini-css-extract-plugin >=2.7.3 development
- minimatch >=7.4.2 development
- monaco-editor-webpack-plugin >=7.0.1 development
- npm >=9.6.1 development
- pnp-webpack-plugin >=1.7.0 development
- postcss-flexbugs-fixes >=5.0.2 development
- postcss-loader >=7.0.2 development
- postcss-normalize >=10.0.1 development
- postcss-preset-env >=8.0.1 development
- react-dev-utils >=12.0.1 development
- resolve >=1.22.1 development
- sass-loader >=13.2.0 development
- style-loader >=3.3.2 development
- typescript >=4.9.5 development
- url-loader >=4.1.1 development
- webpack >=5.76.1 development
- webpack-dev-server >=4.12.0 development
- webpack-manifest-plugin >=5.0.0 development
- workbox-webpack-plugin >=6.5.4 development
- @fluentui/react ^8.106.6
- @uifabric/react-hooks ^7.16.4
- axios ^1.3.4
- copy-to-clipboard ^3.3.3
- d3 ^7.8.2
- echarts ^5.4.1
- echarts-for-react ^3.0.2
- json5 ^2.2.3
- parcoord-es ^2.2.10
- rc-progress ^3.4.1
- react ^17.0.2
- react-dom ^17.0.2
- react-json-tree ^0.18.0
- react-monaco-editor ^0.52.0
- react-paginate ^8.1.4
- react-router-dom ^6.9.0
dependencies/required.txt
pypi
- PythonWebHDFS *
- astor *
- cloudpickle *
- colorama *
- filelock <3.12
- json_tricks >=3.15.5
- numpy *
- numpy <1.22
- nvidia-ml-py *
- packaging *
- pandas *
- prettytable *
- psutil *
- pyyaml >=5.4
- requests *
- responses *
- schema *
- scikit-learn >=0.24.1
- scipy <1.8
- scipy *
- tqdm *
- typeguard >=3.0.0
- typing_extensions >=4.0.0
- websockets >=10.1
dependencies/required_extra.txt
pypi
- ConfigSpace >=0.4.17
- ConfigSpaceNNI >=0.4.7.3
- gym *
- hyperopt ==0.1.2
- pybnn *
- smac4nni *
- statsmodels >=0.12.0
examples/nas/benchmarks/nasbench101.requirements.txt
pypi
- peewee *
- tensorflow *
- tqdm *
examples/nas/benchmarks/nasbench201.requirements.txt
pypi
- gdown *
- peewee *
- torch *
- tqdm *
examples/nas/benchmarks/nds.requirements.txt
pypi
- peewee *
- tqdm *
examples/nas/benchmarks/nlp.requirements.txt
pypi
- peewee *
examples/trials/benchmarking/automlbenchmark/requirements.txt
pypi
- matplotlib >=3.4.1
- pandas >=1.2.0
- pyyaml >=5.4.1
examples/trials/ga_squad/requirements.txt
pypi
- tensorflow *
examples/trials/mnist-batch-tune-keras/requirements.txt
pypi
- keras >=2.2.4
examples/trials/mnist-pytorch/requirements.txt
pypi
- torch *
- torchvision *
examples/trials/mnist-sharedstorage/requirements.txt
pypi
- torch *
- torchvision *
examples/trials/network_morphism/requirements.txt
pypi
- Keras *
- numpy *
- tensorflow *
- torch *
- torchvision *
examples/trials/sklearn/requirements.txt
pypi
- python3-mpipinstall--usernumpyscipysklearn *
- sudoapt-getinstalllibblas-devliblapack-devlibatlas-base-devgfortran *
examples/tuners/customized_tuner/setup.py
pypi
nni/algorithms/feature_engineering/gbdt_selector/requirements.txt
pypi
- lightgbm *
nni/algorithms/feature_engineering/gradient_selector/requirements.txt
pypi
- numpy *
- scikit-learn *
- scipy *
- torch *
setup.py
pypi