Saved in:
| Main Authors: | , , |
|---|---|
| Format: | Preprint |
| Published: |
2026
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2606.01114 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866913176954601472 |
|---|---|
| author | Cheng, Tsz Chung Kurokawa, Yuichiro Yuasa, Hiromi |
| author_facet | Cheng, Tsz Chung Kurokawa, Yuichiro Yuasa, Hiromi |
| contents | Micromagnetic simulations are essential tools in nanomagnetism and spintronics research. Although widely adopted solvers like Mumax3 and the Python-native magnum.np use GPU acceleration to improve performance, these tools are limited to single-device computation. In this work, we present the first Python-native multi-GPU micromagnetic framework by extending magnum.np with PyTorch Distributed. This leverages high-speed communication and computation across multiple GPUs while retaining the benefits of ease of installation, platform-agnostic design, and compatibility with Python. For computationally intensive demagnetisation effective-field calculations, we achieve a 7.0x speedup across 8 GPUs connected via NVLink, whereas Halo exchange required for Heisenberg exchange shows limited scaling due to kernel dispatch latency. We also demonstrated the framework's versatility by achieving a 6.8x speedup in demagnetisation field computation on CPU with NUMA pinning via the MPI backend of PyTorch Distributed. Faster turnaround times will enable researchers to explore larger, more complex systems and accelerate the design cycle for novel spintronic devices. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2606_01114 |
| institution | arXiv |
| publishDate | 2026 |
| record_format | arxiv |
| spellingShingle | Magnum.np.distributed: Accelerating Finite Difference Micromagnetic Simulations with Multiple GPUs Cheng, Tsz Chung Kurokawa, Yuichiro Yuasa, Hiromi Distributed, Parallel, and Cluster Computing Mesoscale and Nanoscale Physics Micromagnetic simulations are essential tools in nanomagnetism and spintronics research. Although widely adopted solvers like Mumax3 and the Python-native magnum.np use GPU acceleration to improve performance, these tools are limited to single-device computation. In this work, we present the first Python-native multi-GPU micromagnetic framework by extending magnum.np with PyTorch Distributed. This leverages high-speed communication and computation across multiple GPUs while retaining the benefits of ease of installation, platform-agnostic design, and compatibility with Python. For computationally intensive demagnetisation effective-field calculations, we achieve a 7.0x speedup across 8 GPUs connected via NVLink, whereas Halo exchange required for Heisenberg exchange shows limited scaling due to kernel dispatch latency. We also demonstrated the framework's versatility by achieving a 6.8x speedup in demagnetisation field computation on CPU with NUMA pinning via the MPI backend of PyTorch Distributed. Faster turnaround times will enable researchers to explore larger, more complex systems and accelerate the design cycle for novel spintronic devices. |
| title | Magnum.np.distributed: Accelerating Finite Difference Micromagnetic Simulations with Multiple GPUs |
| topic | Distributed, Parallel, and Cluster Computing Mesoscale and Nanoscale Physics |
| url | https://arxiv.org/abs/2606.01114 |