Skip to main content

magnum.np finite-difference package for the solution of micromagnetic problems

Project description

magnum.np

magnum.np 2.0.1

magnum.np is a Python library for the solution of micromagnetic problems with the finite-difference method. It implements state-of-the-art algorithms and is based on pytorch, which allows to seamlessly run code either on GPU or on CPU. Simulation scripts are written in Python which leads to very readable yet flexible code. Due to pytorch integration, extensive postprocessing can be done directly in the simulations scripts. Furthermore pytorch's autograd feature makes it possible to solve inverse problems without significant modifications of the code. This manual is meant to give you both a quick start and a reference to magnum.np.

Version 2.0: The magnum.np interface slightly changed since version 2.0. Find more details about the necessary changes and the motivation here.

Features

  • Explicit / Implicit time-integration of the Landau-Lifshitz-Gilbert Equation
  • Fast FFT Demagnetization-field computation optimized for small memory footprint
  • Fast FFT Oersted-field optimized for small memory footprint
  • Fast FFT Vector-Potential optimized for small memory footprint
  • Periodic Boundary Conditions in 1D, 2D, and 3D (True and Pseudo-Periodic)
  • Non-Equidistant Mesh for Multilayer Structures
  • Arbitrary Material Parameters varying in space and time
  • Spin-torque model by Slonczewski
  • Spin-torque model by Zhang and Li
  • Spin-Orbit torque (SOT)
  • Bilinear and biquadratic Antiferromagnetic coupling layers (RKKY)
  • Dzyaloshinskii-Moriya interaction (interface, bulk, D2d)
  • String method for energy barrier computations
  • Eigenmode Solver for efficient calculation of normal modes
  • Sophisticated domain handling, e.g. for spatially varying material parameters
  • efficient Voronoi Code for 2D and 3D problems (including intergrain phase)
  • Seamless VTK import / export via pyvista
  • Inverse Problems via pytorch's autograd feature

Documented Demos:

Demo scripts for various applications are available in the demo directory.

The following demos are also stored on Google Colab, where they can directly be run without any local installation:

Installation

For a clean and independent system, we start with a clean virtual python environment (this step could be omitted, if you would like to install magnum.np into the global python environment)

mkdir venv
python -m venv venv
source venv/bin/activate

from Python Package Index (PyPi)

In order to install a release versions of magnum.np one simple uses:

pip install magnumnp

You can also easily install different versions from private repositories. E.g. use the following command to install the latest version of the main branch:

pip install git+https://gitlab.com/magnum.np/magnum.np@main

from source code (gitlab.com)

More advanced users can also install magnum.np from source code. It can be downloaded from https://gitlab.com/magnum.np/magnum.np .

After activating the virtual environment magnum.np can be installed using the pip -e option which allows to easily modify the source code:

pip install -e .

Note that a default version of pytorch is included in magnum.np's dependecy list. If you would like to uses a specific pytorch version (fitting your installed CUDA library) it needs to be installed in advance.

run remotely via Google Colab


Magnum.np could also be used without any hardware by executing it remotely on resources provided by Google Colab. The platform offers different runtime types like CPU(None), GPU or TPU. This allows users to directly test magnum.np, whithout needing their own hardware. Advanced users can use Google Colab(Pro), which provides access to current GPUs like the A100.

Some jupyter-notebook examples are included in the demo directory, which also include links to Colab, where they can directly be run without any local installation.

Example

The following demo code shows the solution of the muMAG Standard Problem #5 and can be found in the demos directory:

from magnumnp import *
import torch

Timer.enable()

# initialize state
n  = (40, 40, 1)
dx = (2.5e-9, 2.5e-9, 10e-9)
mesh = Mesh(n, dx)

state = State(mesh)
state.material = {
    "Ms": 8e5,
    "A": 1.3e-11,
    "alpha": 0.1,
    "xi": 0.05,
    "b": 72.17e-12
    }

# initialize magnetization that relaxes into s-state
state.m = state.Constant([0,0,0])
state.m[:20,:,:,1] = -1.
state.m[20:,:,:,1] = 1.
state.m[20,20,:,1] = 0.
state.m[20,20,:,2] = 1.

state.j = state.Tensor([1e12, 0, 0])

# initialize field terms
demag    = DemagField()
exchange = ExchangeField()
torque   = SpinTorqueZhangLi()

# initialize sstate
llg = LLGSolver([demag, exchange])
llg.relax(state)
write_vti(state.m, "data/m0.vti", state)

# perform integration with spin torque
llg = LLGSolver([demag, exchange, torque])
logger = ScalarLogger("data/m.dat", ['t', 'm'])
while state.t < 5e-9:
    llg.step(state, 1e-10)
    logger << state

Timer.print_report()

Documentation

The documentation is located in the doc directory and can be built using sphinx. For example the following commands build an HTML documentation of the actual source code and stores it in the public folder:

sphinx-build -b html docs public

Alternatively, the latest version of the documentation is always available on https://magnum.np.gitlab.io/magnum.np/

Citation

If you use magnum.np in your work or publication, please cite the following reference:

[1] Bruckner, Florian, et al. "magnum.np -- A pytorch based GPU enhanced Finite Difference Micromagnetic Simulation Framework for High Level Development and Inverse Design", to be published (2023).

Contributing

Contributions are gratefully accepted. The source code is hosted on www.gitlab.com/magnum.np/magnum.np. If you have any issues or question, just open an issue via gitlab.com. To contribute code, fork our repository on gitlab.com and create a corresponding merge request.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

magnumnp-2.0.1.tar.gz (74.4 kB view details)

Uploaded Source

Built Distribution

magnumnp-2.0.1-py3-none-any.whl (108.9 kB view details)

Uploaded Python 3

File details

Details for the file magnumnp-2.0.1.tar.gz.

File metadata

  • Download URL: magnumnp-2.0.1.tar.gz
  • Upload date:
  • Size: 74.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.10.15

File hashes

Hashes for magnumnp-2.0.1.tar.gz
Algorithm Hash digest
SHA256 43dd5440a20641429d4dcf9df72ae33c1246ba933945e9154853c8fb48ac7823
MD5 8030c27d1bf79dd6a151067459b23081
BLAKE2b-256 8f050dc5180fb15ba5b6fe2d69bd374510778324383e51e453d0536ca5ef61b0

See more details on using hashes here.

File details

Details for the file magnumnp-2.0.1-py3-none-any.whl.

File metadata

  • Download URL: magnumnp-2.0.1-py3-none-any.whl
  • Upload date:
  • Size: 108.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.10.15

File hashes

Hashes for magnumnp-2.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 73b611561763d2783cce34848737b84ff68ef8b70a16fdc98c9abc0c0ddc31eb
MD5 2bf5494febc361264a86c0c7e664c4e3
BLAKE2b-256 ff5400e96731d339d1504be0982dcefd6da02a1271bd440a694f7368e277be9d

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page