DistML provide a supplement to mllib to support model-parallel on Spark - View it on GitHub