[DataLoader] Add simple `groupby` DataPipe (#60675)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60675
Test Plan: Imported from OSS
Reviewed By: mruberry
Differential Revision: D29461082
Pulled By: VitalyFedyunin
fbshipit-source-id: ded5a3a1555bfd8457d64b7e61ab6729fff9cb75
