Merge branch 'kmeans-em-refactor' into 'master'

[kmeans] refactor e-m to have simpler graphs for dask

Closes #36

See merge request !48