Google ADK might be useful, especially v2 reorients it around graph operators for control flow.

Your specific case is listed in the v2 docs with an operation that fans out to parallel many tasks then joins the results.