Like in `GraphFrames`: https://github.com/graphframes/graphframes/blob/master/core/src/main/scala/org/graphframes/lib/AggregateMessages.scala#L60
