This is the same as Spark, where we can fuse stages with narrow dependencies to take advantages of the producer-consumer locality.