JackBrady

object-centric-library

Library for the training and evaluation of object-centric models (ICML 2022)

Transformer-Encoder

Implementation of Transformer encoder in PyTorch

ima-vae

This is the code for the paper Embrace the Gap: VAEs perform Independent Mechanism Analysis, showing that optimizing the ELBO is equivalent to optimizing the IMA-regularized log-likelihood under certain assumptions (e.g., small decoder variance).

objects-compositional-generalization

Official code for the paper "Provable Compositional Generalization for Object-Centric Learning" (ICLR 2024, oral)

objects-identifiability