flax_model/alphafold3/structure/cpp/aggregation.pyi

from collections.abc import Sequence

def indices_grouped_by_value(values: Sequence[int]) -> dict[int, list[int]]: ...