where η is the learning rate and S p represents the update step function of the given optimizer. Various packages implementing differentiable molecular dynamics are available, such as JAX-MD, TorchMD, and Molly. However, these programs lack some critical features, especially the particle mesh routines needed for the HhPF nonbonded interactions. Therefore, we have implemented our differentiable framework in ∂-HyMD