Micromagnetic realization of energy-based models using stochastic magnetic tunnel junctions

Energy-based models (EBMs) can bridge physics, machine learning, and statistics. EBMs provide a unified and powerful platform to describe, learn, and optimize complex systems. In this paper, we propose a neuromorphic implementation of EBMs using a network of stochastic magnetic tunnel junctions (MTJs) that can perform energy minimization and solve optimization problems. Our implementation builds on the Object Oriented MicroMagnetic Framework (OOMMF). We derive the different energy terms and map them to the micromagnetic Landau-Lifshitz-Gilbert (LLG) equation. We then develop a C + + module for EBMs which integrates seamlessly with OOMMF. We demonstrate our implementation on a full set of logic gates using stochastic MTJs networks. Our method offers several advantages, including fast modeling of EBMs with spintronic devices and design insights for stochastic MTJ-based neuromorphic circuits.