Stars
Fast inference from large lauguage models via speculative decoding
Spec-Bench: A Comprehensive Benchmark and Unified Evaluation Platform for Speculative Decoding (ACL 2024 Findings)
[CVPR 2025 Oral] PyTorch re-implementation for Autoregressive Distillation of Diffusion Transformers (ARD).
Official PyTorch implementation for Distilling Dataset into Neural Field [ICLR 2025]
(Official) PyTorch implementation for Trajectory-Class-Aware Multi-Agent Reinforcement Learning (ICLR 2025)
[ICLR 2025 Spotlight] Official PyTorch implementation for Diffusion Bridge AutoEncoders for Unsupervised Representation Learning (DBAE).
Official PyTorch implementation for Reward-based Input Construction for Cross-document Relation Extraction (REIC) in ACL 2024.
Official PyTorch implementation For Sharpness-Aware Active Learning [ICML 2023]
Official PyTorch implementation for Label-Noise Robust Diffusion Models (TDSM) in ICLR 2024.
About Official PyTorch implementation for Training Unbiased Diffusion Models From Biased Dataset in ICLR 2024.
Code release for ICLR 2024 paper: Exploring Diffusion Time-steps for Unsupervised Representation Learning
Latent Point Diffusion Models for 3D Shape Generation
Flow-based generative model for 3D point clouds.
Official PyTorch implementation for Training Unbiased Diffusion Models From Biased Dataset in ICLR 2024.
Official PyTorch implementation for Label-Noise Robust Diffusion Models (TDSM) in ICLR 2024.
Official PyTorch implementation for paper: Diffusion-GAN: Training GANs with Diffusion
Official PyTorch implementation for Make Prompts Adaptable: Bayesian Modeling for Vision-Language Prompt Learning with Data-Dependent Prior [AAAI 2024]
Official PyTorch implementation for Frequency Domain-based Dataset Distillation [NeurIPS 2023]
Official Tensorflow implementation for Deep Generative Positive-Unlabeled Learning under Selection Bias (VAE-PU) in CIKM 2020.
Official PyTorch implementation for Multi-modal Text Recognition Networks: Interactive Enhancements between Visual and Semantic Features (MATRN) in ECCV 2022.
Official repo for consistency models.
Official repo for Discriminator Guidance for ImageNet256.