Improve grouping feature in EmbeddingsWriter

[nkaenzig]

For slide level tasks, we should use group_key to group patches from the same slide into the same .pt file(s).
The way it is implemented now still generates one file per patch. We probably should accumulate patches in memory over multiple batches before saving them to file.