msmJAX: Fast and differentiable electrostatics on the GPU in Python. - Florian Buchner; Johannes Schörghuber; Nico Unglert; Jesús Carrete; Georg K. H. Madsen (Comput. Phys. Commun., 2026)