IBM Analog Hardware Acceleration Kit: A Flexible and Fast PyTorch Toolkit for Simulating ANN Training and Inference on Resistive Crossbar Arrays

Speaker: Malte Rasch, IBM

Memristive crossbar arrays are a promising future technology for accelerating AI workloads, but noise and non-idealities demand for improved algorithmic solutions. We introduce the IBM Analog Hardware Acceleration Kit, a first of a kind open source toolkit to simulate crossbar arrays from within PyTorch, to conveniently estimate the impact of material properties and non-idealities on the accuracy for arbitrary ANNs.