This is a model repository for my paper Invariant Slot Attention. I uploaded a single checkpoint for each dataset + model variant I used in the paper except for Waymo Open trained with depth targets.

Code (JAX/FLAX): https://github.com/google-research/google-research/tree/master/invariant_slot_attention

Paper (ICML'23): https://arxiv.org/abs/2302.04973

Demo: https://huggingface.co/spaces/ondrejbiza/isa

I used the TPU Research Cloud to train these models.