nni

模型调参交给机器自动试,最优结果坐等出炉

微软开源的 AutoML 工具包,覆盖机器学习生命周期的自动化,包括特征工程、神经网络架构搜索、模型压缩与超参调优等环节。借助丰富的搜索算法与可视化面板,帮助团队在有限的算力预算下高效找到更优模型,是自动化建模的得力助手。

开源 free 基础设施
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 14363
维护状态 低维护
是否开源
定价模式 free

项目数据

分类基础设施
开发团队microsoft
所属国家
官网地址
定价模式free
价格说明开源工具,MIT许可证,完全免费,无付费版本。
访问状态
是否开源
开源协议MIT
主要语言Python
技术栈/模型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
GitHub 星标★ 14363
30天Star增速
HF 下载量
上线时间2018-06-01 00:00:00
最近更新2026-09-22 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数13

使用教程

难度:入门 约 10 分钟 部署方式:库/依赖 4 步

环境要求

  • 已安装 Python 及配套的 pip(README 未指定具体版本,请以官方安装指南为准)
  • 网络可访问 PyPI,能够执行 pip 安装
  • 如需了解完整环境要求与源码构建方式,请参阅官方安装指南 https://nni.readthedocs.io/en/stable/installation.ht

安装与启动步骤

  1. 1确认 Python 环境

    README 未指定 Python 版本,请先确认本机已装好 Python 与 pip,可执行 pip 命令。

  2. 2安装 NNI

    README 给出的当前版本安装方式,直接通过 pip 从 PyPI 安装 nni 包。

    pip install nni
  3. 3升级到最新版

    如需把已安装的 NNI 更新到最新版本,在安装命令后追加 --upgrade 参数。

    pip install --upgrade nni
  4. 4查阅文档与示例

    README 指向官方文档,内含最新特性、API、示例与教程,安装后按文档继续使用。

如何确认成功

安装完成后,在 Python 环境中可以正常导入 nni 即表示安装成功(README 未给出专门的验证命令)。

常见问题

Q:除了 pip 安装还有别的方式吗?

A:有。README 说明既可以通过 pip 安装,也可以从源码构建,具体步骤见官方安装指南 https://nni.readthedocs.io/en/stable/installation.html。

Q:如何把已安装的 NNI 更新到最新版?

A:在安装命令上追加 --upgrade 参数,即执行 pip install --upgrade nni。

Q:有中文文档吗?

A:有。README 提供了简体中文版文档入口:https://nni.readthedocs.io/zh/stable 。

Q:NNI 能做什么?

A:NNI 自动化特征工程、神经网络架构搜索、超参调优和深度学习模型压缩,覆盖机器学习生命周期多个环节。

注意事项

  • README 只给出了 pip 安装与源码构建两种方式的指引,未给出端口、路径、配置文件等参数,请以官方文档为准。
  • NNI 官方已发布 v3.0 preview(v3.0rc1),如需体验新特性请参考 releases 页面。
  • README 未提供启动服务或运行示例的具体命令,安装后请按官方文档与教程操作。

核心亮点

  • 提供完整的AutoML生命周期管理,覆盖特征工程、神经架构搜索、模型压缩和超参调优,功能全面
  • 支持分布式训练和多框架集成(如PyTorch、TensorFlow),可扩展性强
  • 拥有可视化Web界面,便于实验管理和结果对比,提升用户体验

不足之处

  • 文档/社区待观察
  • 项目维护活跃度近年下降,新功能迭代放缓

适用场景

  • 企业级机器学习平台建设,需要统一管理多种AutoML任务
  • 研究团队进行神经架构搜索和模型压缩实验
  • 数据科学家快速调优模型超参数,提升建模效率

替代项目

Optuna、Ray Tune、AutoKeras

项目介绍

nni 是模型训练领域的开源项目,由 microsoft 开发,2018 年首次发布。

在全站 13,014 个收录项目中,它的 GitHub 星标数(14,363)位列前 4%,在模型训练分类的 518 个项目里位列前 3%。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-22。开源工具,MIT许可证,完全免费,无付费版本。

它主要面向的使用场景是:企业级机器学习平台建设,需要统一管理多种AutoML任务。同类可对比的替代方案包括 Optuna、Ray Tune、AutoKeras。

上一篇:agent-lightning

下一篇:wandb

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13