Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".

OpenXAI : Towards a Transparent Evaluation of Model Explanations

This repository contains data readers and examples for the three tracks of the Shifts Dataset and the Shifts Challenge.

🔬 Research Framework for Single and Multi-Players 🎰 Multi-Arms Bandits (MAB) Algorithms, implementing all the state-of-the-art algorithms for single-player (UCB, KL-UCB, Thompson...) and multi-play…

The official implementation of "The Shapley Value of Classifiers in Ensemble Games" (CIKM 2021).

Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"

Flax is a neural network library for JAX that is designed for flexibility.

[ICML 2023] The official implementation of the paper "TabDDPM: Modelling Tabular Data with Diffusion Models"

A latent text-to-image diffusion model

AIMET is a library that provides advanced quantization and compression techniques for trained neural network models.

An open source implementation of CLIP.

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

Code for the paper "ViperGPT: Visual Inference via Python Execution for Reasoning"

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

Fast and memory-efficient exact attention

Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).