Stars
Official Implementation for the paper "d1: Scaling Reasoning in Diffusion Large Language Models via Reinforcement Learning"
Enhancing Reasoning For Diffusion LLMs via Distribution Matching Policy Optimization
An open-source toolbox for fast sampling of diffusion models. Official implementations of our works published in ICML, NeurIPS, CVPR.
Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).
Official PyTorch implementation for "Large Language Diffusion Models"
Pytorch implementation for MeanFlow
This repository is the official implementation of the paper: Physics Informed Distillation for Diffusion Models, accepted by Transactions on Machine Learning Research (TMLR).
📖 A curated list of resources dedicated to hallucination of multimodal large language models (MLLM).
Official repo for Understanding and Mitigating Memorization in Generative Models via Sharpness of Probability Landscapes [ICML 25]
[CVPR 2025] Official PyTorch implementation of Not All Parameters Matter: Masking Diffusion Models for Enhancing Generation Ability
[NeurIPS 2024] Source code for our paper "Finding NeMo: Localizing Neurons Responsible For Memorization in Diffusion Models".
[CVPR 2025] FedBiP: Heterogeneous One-Shot Federated Learning with Personalized Latent Diffusion Models
SANA: Efficient High-Resolution Image Synthesis with Linear Diffusion Transformer
[NeurIPS 2024 D&B Track] UnlearnCanvas: A Stylized Image Dataset to Benchmark Machine Unlearning for Diffusion Models by Yihua Zhang, Chongyu Fan, Yimeng Zhang, Yuguang Yao, Jinghan Jia, Jiancheng …
Official implementation of NeurIPS'24 paper "Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models". This work adversarially unlearns the text encoder to enh…
A repository of resources on machine unlearning for diffusion models
The official implementation of ECCV'24 paper "To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images ... For Now". This work introduces one fast and e…
[ICLR24 (Spotlight)] "SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation" by Chongyu Fan*, Jiancheng Liu*, Yihua Zhang, Eric Wong, D…
[ICML25] Official repo for "Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond"
[ICML 2025] Unlearning in Diffusion Models using Sparse Autoencoders
Unofficial implementation of "Simplifying, Stabilizing & Scaling Continuous-Time Consistency Models" for MNIST
Official implementation of "A simple, efficient and scalable contrastive masked autoencoder for learning visual representations".
PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
Elucidating the Design Space of Diffusion-Based Generative Models (EDM)