A flexible and fast PyTorch toolkit for simulating training and inference on analog crossbar arrays

Malte J. Rasch,Diego Moreda,Tayfun Gokmen,Manuel Le Gallo,Fabio Carta, Cindy Goldberg,Kaoutar El Maghraoui,Abu Sebastian,Vijay Narayanan

2021 IEEE 3rd International Conference on Artificial Intelligence Circuits and Systems (AICAS)（2021）

引用 46|浏览5

暂无评分

摘要

We introduce the IBM ANALOG HARDWARE ACCELERATION KIT, a new and first of a kind open source toolkit to simulate analog crossbar arrays in a convenient fashion from within PYTORCH (freely available at https://github.com/IBM/aihwkit). The toolkit is under active development and is centered around the concept of an “analog tile” which captures the computations performed on a crossbar array. Analog t...

查看译文

关键词

Training,Tiles,Software algorithms,Artificial neural networks,Tools,Programming,Inference algorithms

AI 理解论文

溯源树

样例

生成溯源树，研究论文发展脉络

Chat Paper

正在生成论文摘要