In Fig 1, we also show the difference between the loss function obtained by training a DPPC membrane with and without the PRS method. Without PRS, the loss is noisy and slowly decreasing, meaning that, due to the short single differentiable MD simulation, we are not able to consistently learn the parameters.