The reviewers and I agree on the contribution of this paper. However, there are still some concerns and suggestions that would extremely benefit the quality of this submission. The authors are strongly encouraged to address these points before the camera ready paper: - Please add more discussions on existing works on the Stein score function. - Please aim to go beyond two variables - Please do include results or a discussion on cross entropy