maderix

ANE

maderix

Training neural networks on Apple Neural Engine via reverse-engineered private APIs

AI 简介

该项目通过逆向工程的私有API直接在苹果神经引擎(ANE)上训练神经网络。其核心功能是利用Objective-C语言实现对ANE的直接访问,绕过了Apple限制ANE仅用于推理的软件屏障,展示了ANE在训练任务中的潜力。项目包括了基于_ANEClient和_ANECompiler私有API的概念验证、一系列记录ANE实际性能特征的基准测试以及对于希望探索直接ANE访问的研究人员的参考。尽管目前硬件利用率较低且存在许多技术挑战,如部分元素级操作仍需回退到CPU执行,但该项目非常适合于研究者探索ANE及其他类似神经处理单元(NPU)在非生产环境下的训练能力。

Objective-C
MIT License
6.7k
Stars
926
Forks
65
Watchers
9
Issues

Star 增长

今日+17
近 7 天+19
近 30 天+52
综合评分99.6
默认分支main