you either wanna groupBy and then apply another transform to the resulting map to make it a

Map<A, MutableList<B>>