True, `ModelDrivenTransform` should just ask for the Jacobian as you stated.

Not sure about the shape of the Jacobian though, I thought we said it should be `n_points, n_dims, N-params`. Remember p are scalars here...