ai8x-training

在电脑上训练量化模型，一键部署到毫瓦级 AI 芯片

开源 🏷️完全免费 🔧 仅开源项目，无在线服务，需自行部署 模型训练 🍴 复刻 103 👀 关注 17
GitHub ↗
GitHub 星标 ★ 117
维护状态 维护中
是否开源 开源
定价模式 🏷️完全免费 开源免费，需自行部署训练

项目预览

ai8x-training 项目界面预览图
图片来源：GitHub 官方仓库名片

项目数据

分类模型训练
开发团队analogdevicesinc
GitHubhttps://github.com/analogdevicesinc/ai8x-training
定价模式🏷️完全免费
价格说明开源免费，需自行部署训练
访问状态🔧 仅开源项目，无在线服务，需自行部署
是否开源是
开源协议Apache-2.0
主要语言Jupyter Notebook
技术栈/模型ai,analog-devices,artificial-intelligence,deep-learning,machine-learning,max78000,max78002,maxim,maxim-integrated
GitHub 星标★ 117
上线时间2020-05-19 00:00:00
代码更新2026-08-24
仓库体积222.2 MB
本站数据更新2026-11-05
维护状态维护中
收录时间2026-09-19
浏览次数1

项目介绍

ai8x-training 是亚德诺半导体（ADI，原 Maxim）为 MAX78000 与 MAX78002 两款超低功耗边缘 AI 微控制器提供的官方模型训练工具链。它解决的核心问题是：这类芯片采用 CNN 加速器，对网络结构、量化方式和数据布局有严格约束，普通 PyTorch 模型无法直接部署。项目基于 PyTorch 构建，提供量化感知训练（QAT）流程、预置的 ai8x 模型库、数据集加载与合成数据生成、逐层权重与偏置导出，以及与 ai8x-synthesis 工具衔接的检查点格式。开发者可在 PC 上完成 1/2/4/8 位权重量化训练，再交给配套编译器生成芯片可执行代码，从而把关键词识别、图像分类、人脸检测等任务跑在毫瓦级功耗的 MCU 上。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化，观测越久曲线越完整

数据来源：GitHub API 逐日快照 · 收录于 2026-09-19 · 当前星标 ★ 117

核心亮点

  • 提供完整的量化感知训练流程，支持 1/2/4/8 位权重，直接对应 MAX78000/MAX78002 硬件加速器位宽
  • 内置大量经过验证的参考模型（如 KWS20、MNIST、CIFAR-100、人脸检测），可作迁移学习起点
  • 与 ai8x-synthesis 无缝衔接，训练产物可直接编译为芯片固件，端到端链路清晰

不足之处

  • 强绑定 ADI 自家芯片，模型结构和算子受限，无法用于其他硬件平台
  • 社区规模小、星标仅百余，第三方教程和问题解答较少，遇到坑主要靠官方文档

适用场景

  • 为 MAX78000 开发关键词唤醒（KWS）固件，训练低功耗语音命令模型
  • 在电池供电的视觉传感器上做简单图像分类或人脸检测
  • 高校/企业评估超低功耗边缘 AI 方案时，快速跑通训练到部署的完整链路

替代项目

TensorFlow Lite for Microcontrollers、Edge Impulse、CMSIS-NN

上一篇：litData

下一篇：long-context-attention

同类项目推荐

Composition-RL 开源

把可验证提示词自动组合，低成本扩充 RL 训练数据

[Neurips 2026] Official repository for the paper "Composition-RL: Compose Your Verif···

★ 138 2026-10-05
DataMate 开源

一站式搞定大模型微调与RAG的数据处理，让数据准备不再拖后腿

DataMate is an enterprise-level data processing platform designed for model fine-tun···

★ 374 2026-08-09
nntrainer 开源

让手机家电自己训练模型，数据不出设备

NNtrainer is Software Framework for Training and Inferencing Neural Network Models o···

★ 215 2026-10-17
paradigms 开源

并行采样加速扩散模型，少步生成高质量图像

PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spo···

★ 159 2026-08-09
neptune-client 开源

一键记录训练实验，自动追踪指标与模型，团队协作更高效

The experiment tracker for foundation model training

★ 622 2026-08-09
meta-context-engineering 开源

让AI自动进化上下文策略，告别手写提示词

[ICML 2026] Meta Context Engineering via Agentic Skill Evolution

★ 185 2026-08-09
privacy 开源

给模型训练加隐私锁，敏感数据也能放心训

Library for training machine learning models with privacy for training data

★ 2038 2026-10-17
Simple-Lora 开源

跟着中文教程，训练你的专属虚拟偶像 LoRA

diffusion lora chinese tutorial，虚拟idol训练中文教程

★ 81 2026-08-09