Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

An open source implementation of CLIP.

Release for Improved Denoising Diffusion Probabilistic Models

Prompt Learning for Vision-Language Models (IJCV'22, CVPR'22)

Trustworthy AI related projects

PyTorch implementation of normalizing flow models

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

The Official PyTorch Implementation of "LSGM: Score-based Generative Modeling in Latent Space" (NeurIPS 2021)

Code for Diff-SCM paper

Code for the paper Open-Vocabulary Attention Maps with Token Optimization for Semantic Segmentation in Diffusion Models @ CVPR 2024

[ICLR 2025] VL-ICL Bench: The Devil in the Details of Multimodal In-Context Learning

Code repository of the paper "CITRIS: Causal Identifiability from Temporal Intervened Sequences" and "iCITRIS: Causal Representation Learning for Instantaneous Temporal Effects"

VAEs and nonlinear ICA: a unifying framework

[ACM MM 2022] Towards Counterfactual Image Manipulation via CLIP

Official implementation of our EMNLP 2022 paper "CPL: Counterfactual Prompt Learning for Vision and Language Models"

Neural Causal Model (NCM) implementation by the authors of The Causal Neural Connection.