msmJAX: Fast and Differentiable Electrostatics on the GPU in Python

Florian Buchner, Johannes Schörghuber, Nico Unglert +2

We present msmJAX, a Python package implementing the multilevel summation method with B-spline interpolation, a linear-scaling algorithm for efficiently evaluating electrostatic an…