@jalabort this is the point I propose to handle reshaping of the jacobians. `ModelDrivenTransform` should just call

    dX_dp = self.pdm.d_dp(points)

and expect back something of shape `(n_points, n_params, n_dims)`