Support distributed implementation with sumpy