SALMONN: Speech Audio Language Music Open Neural Network

(Crafter + NetHack) in JAX. ICML 2024 Spotlight.

An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.

All graph/GNN papers accepted at the International Conference on Machine Learning (ICML) 2024.

Advances on machine learning of graphs, covering the reading list of recent top academic conferences.

[ICML 2024] See More Details: Efficient Image Super-Resolution by Experts Mining

[ICML 2024] Let Go of Your Labels with Unsupervised Transfer

JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"

[ICML 2024] Language Models Represent Beliefs of Self and Others

🔥OSN in PyTorch (ICML 2024)

Python library for multipersistence

Official PyTorch implementation of the ICML 2024 paper "Hyperbolic Active Learning for Semantic Segmentation under Domain Shift"

[ICML 2024] Outlier-Efficient Hopfield Layers for Large Transformer-Based Models

🐼PANDA: Expanded Width-Aware Message Passing Beyond Rewiring, ICML 2024

Source code of the ICML24 paper "Self-Composing Policies for Scalable Continual Reinforcement Learning" (selected for oral presentation)

Official code for "Enabling Uncertainty Estimation in Iterative Neural Networks" (ICML 2024)

Official PyTorch implementation of "Stochastic Conditional Diffusion Models for Robust Semantic Image Synthesis" (ICML 2024).

Official code for "ZigZag: Universal Sampling-free Uncertainty Estimation Through Two-Step Inference" (TMLR 2024)

The code repository for ICML24 paper "Tabular Insights, Visual Impacts: Transferring Expertise from Tables to Images"

Compositional Capabilities of Autoregressive Transformers: A Study on Synthetic, Interpretable Tasks