MagneX: A High-Performance, GPU-Enabled, Data-Driven Micromagnetics Solver for Spintronic Systems

We present MagneX, an open-source micromagnetics modeling tool that leverages massively parallel and GPU-enabled DOE software frameworks and ML workflows to allow for detailed investigations of multiphysics coupling in spintronic devices. We leverage the AMReX framework for multicore and GPU scalability, the SUNDIALS library for high-order multirate time integration, and python-based workflows for data-driven acceleration of computational kernels. MagneX incorporates various crucial magnetic coupling mechanisms, including Zeeman coupling, demagnetization coupling, crystalline anisotropy interaction, exchange coupling, and Dzyaloshinskii-Moriya interaction (DMI) coupling. We demonstrate the performance and scalability of the code and rigorously validate MagneX's functionality using the mumag standard problems and widely-accepted DMI benchmarks. With the capacity to explore complete physical interactions, this innovative approach offers a promising pathway to better understand and develop fully integrated spintronic and electronic systems.