Swift: Best way to handle unique/variable cell spacing in UICollectionView?

Question

Here is an image of iMessage. You can see the red circles show a larger gap between messages than the orange circles.

Say each cell is one of the chat bubbles