The Pix2Code framework: generalizable, interpretable and revisable visual concept learning

Dataset to assess the disentanglement properties of unsupervised learning methods

A library for mechanistic interpretability of GPT-style language models

Must-read Papers on Knowledge Editing for Large Language Models.

Interpretating the latent space representations of attention head outputs for LLMs

Code repository for "The Clock and the Pizza: Two Stories in Mechanistic Explanation of Neural Networks"

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

LLM101n: Let's build a Storyteller

A PyTorch implementation of Model Agnostic Meta-Learning (MAML) that faithfully reproduces the results from the original paper.

Towards Automatic Concept-based Explanations

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

A curated list of awesome Multimodal studies.

The official implementation of Achieving Cross Modal Generalization with Multimodal Unified Representation (NeurIPS '23)

Online Prototype Learning for Online Continual Learning (ICCV 2023)

Visual Concept Connectome

PyTorch implementation of nonsymmetric determinantal point process (DPP) learning.

This is official code repository for "Learning with Mixture of Prototypes for Out-of-Distribution Detection", accepted at ICLR2024

An eXplainable AI toolkit with Concept Relevance Propagation and Relevance Maximization

Code for the paper "Unlocking Slot Attention by Changing Optimal Transport Costs"

Fruit fly neural count sketch

Contrastive Learning for Time Series

Using sparse coding to find distributed representations used by neural networks.

Sparse Autoencoder for Mechanistic Interpretability

Code for the paper "Contrastive Learning Inverts the Data Generating Process".