NeuralMag
An open-source nodal finite-difference micromagnetic code with PyTorch and JAX backends, developed for inverse problems: the simulation is fully differentiable, so geometries and material parameters can be optimized with respect to a target response. Runs on GPUs.