Differentiable neural network representation of multi-well, locally-convex potentials