@mengxr I am not sure of the return type. Indeed, I prefer to use `Vector` in Spark as the return type instead of `BV` in Breeze, but if so, I have to use lots of `toBreeze` in my code.