Official Pytorch Implementation of Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think

Official PyTorch implementation of Holistic Molecular Representation Learning via Multi-view Fragmentation (TMLR 2024)

Official Code for the paper "SuRe: Summarizing Retrievals using Answer Candidates for Open-domain QA of LLMs" (ICLR 2024)

Inference code for CodeLlama models

Collaborative Score Distillation for Consistent Visual Synthesis (NeurIPS 2023)

Official PyTorch implementation of "Differentially Private Sharpness-Aware Training", ICML 2023.

Official PyTorch implementation of STUNT: Few-shot Tabular Learning with Self-generated Tasks from Unlabeled Tables (ICLR 2023 Spotlight).

Official PyTorch implementation of Scalable Neural Video Representations with Learnable Positional Features (NeurIPS 2022).

Meta-Learning with Self-Improving Momentum Target (NeurIPS 2022)

Learning Large-scale Neural Fields via Context Pruned Meta-Learning (NeurIPS 2023)

Official PyTorch implementation of Video Probabilistic Diffusion Models in Projected Latent Space (CVPR 2023).

Preference Transformer: Modeling Human Preferences using Transformers for RL (ICLR2023 Accepted)

PyTorch code accompanying the paper "Imitating Graph-Based Planning with Goal-Conditioned Policies" (ICLR 2023).