magpylib_jax

JAX-native, end-to-end differentiable magnetic field computation with Magpylib-compatible APIs.

magpylib_jax is organized around three goals:

  • keep the high-level Magpylib user model familiar,

  • provide analytical field kernels that remain differentiable through JAX,

  • make correctness and performance observable through parity tests, benchmarks, and profiling artifacts.

Start here

User Guide