Links for slot-attention-pytorch