---
license: mit
---
# Invariant Slot Attention: Object Discovery with Slot-Centric Reference Frames

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

## Acknowledgements

I used the [TPU Research Cloud](https://sites.research.google/trc/about/) to train these models.