the gradient seems to be correct now
Merge branch 'master' of https://github.com/arendu/Featurized-EM