IBM/aihwkit

IBM Analog Hardware Acceleration Kit

解决的问题

IBM Analog Hardware Acceleration Kit (aihwkit) 解决了“冯·诺依曼瓶颈”问题,即在独立的内存和计算单元之间传输数据所造成的效率损失。它支持探索和模拟模拟内存内计算 (AIMC) 硬件,让计算直接在数据存储处进行,从而显著提升 AI 工作负载的速度和能源效率。

运作原理

该工具包提供了一个高性能的 C++ 模拟器 (支持 CUDA),用于建模模拟设备 (如相变内存或 RRAM) 的物理特性以及交叉开关配置。它与 PyTorch 直接集成,允许用户将标准神经网络层替换为模拟版本 (全连接层、卷积层、LSTM),并使用专门的模拟优化器 (如 AnalogSGD)。

目标客群

适合希望模拟模拟 AI 硬件、执行硬件感知训练以增强模型对物理噪声和非理想特性鲁棒性的 AI 研究人员与硬件工程师,或用于探索内存内计算设备的材料规格。

特色亮点

  • PyTorch 集成:在 PyTorch 中无缝使用模拟神经网络模块和训练工作流程。
  • 硬件感知训练:在正向传播中纳入噪声和非理想特性,确保模型能在真实的模拟硬件上运行。
  • 高性能模拟器:支持 CUDA 的 C++ 模拟器,用于建模设备波动、ADC/DAC 离散化以及随机更新脉冲序列。
  • 模型转换:自动将预训练的标准模型转换为等效模拟模型的工具。
  • 广泛的硬件支持:通过 CUDA 支持 NVIDIA GPU,并通过 ROCm/HIP 支持 AMD GPU。

相关

  • 项目
  • 项目
  • Dispatch
  • 项目
  • 项目