GraphIt - A High-Performance DSL for Graph Analytics. - Yunming Zhang; Mengjiao Yang 001; Riyadh Baghdadi; Shoaib Kamil 001; Julian Shun; Saman P. Amarasinghe (CoRR, 2018)

GraphIt: a high-performance graph DSL. - Yunming Zhang; Mengjiao Yang 001; Riyadh Baghdadi; Shoaib Kamil 001; Julian Shun; Saman P. Amarasinghe (Proc. ACM Program. Lang., 2018)

Energy-Based Processes for Exchangeable Data. - Mengjiao Yang 001; Bo Dai 001; Hanjun Dai; Dale Schuurmans (ICML, 2020)

Off-Policy Evaluation via the Regularized Lagrangian. - Mengjiao Yang 001; Ofir Nachum; Bo Dai 001; Lihong Li 001; Dale Schuurmans (NeurIPS, 2020)

Towards Automatic Evaluation of Dialog Systems: A Model-Free Off-Policy Evaluation Approach. - Haoming Jiang; Bo Dai 001; Mengjiao Yang 001; Tuo Zhao; Wei Wei 019 (EMNLP (1), 2021)

Benchmarks for Deep Off-Policy Evaluation. - Justin Fu; Mohammad Norouzi 002; Ofir Nachum; George Tucker; Ziyu Wang 001; Alexander Novikov 001; Mengjiao Yang 001; Michael R. Zhang; Yutian Chen 001; Aviral Kumar; Cosmin Paduraru; Sergey Levine; et al. (ICLR, 2021)

Representation Matters: Offline Pretraining for Sequential Decision Making. - Mengjiao Yang 001; Ofir Nachum (ICML, 2021)

Provable Representation Learning for Imitation with Contrastive Fourier Features. - Ofir Nachum; Mengjiao Yang 001 (NeurIPS, 2021)

Combiner: Full Attention Transformer with Sparse Computation Cost. - Hongyu Ren; Hanjun Dai; Zihang Dai; Mengjiao Yang 001; Jure Leskovec; Dale Schuurmans; Bo Dai 001 (NeurIPS, 2021)

Offline Policy Selection under Uncertainty. - Mengjiao Yang 001; Bo Dai 001; Ofir Nachum; George Tucker; Dale Schuurmans (AISTATS, 2022)

TRAIL: Near-Optimal Imitation Learning with Suboptimal Data. - Mengjiao Yang 001; Sergey Levine; Ofir Nachum (ICLR, 2022)

Marginal Distribution Adaptation for Discrete Sets via Module-Oriented Divergence Minimization. - Hanjun Dai; Mengjiao Yang 001; Yuan Xue; Dale Schuurmans; Bo Dai 001 (ICML, 2022)

Making Linear MDPs Practical via Contrastive Representation Learning. - Tianjun Zhang; Tongzheng Ren; Mengjiao Yang 001; Joseph Gonzalez 001; Dale Schuurmans; Bo Dai 001 (ICML, 2022)

Context-Aware Language Modeling for Goal-Oriented Dialogue Systems. - Charlie Snell; Sherry Yang 001; Justin Fu; Yi Su; Sergey Levine (NAACL-HLT (Findings), 2022)

CHAI: A CHatbot AI for Task-Oriented Dialogue with Offline Reinforcement Learning. - Siddharth Verma; Justin Fu; Sherry Yang 001; Sergey Levine (NAACL-HLT, 2022)

Multi-Game Decision Transformers. - Kuang-Huei Lee; Ofir Nachum; Mengjiao Yang 001; Lisa Lee; Daniel Freeman; Sergio Guadarrama; Ian Fischer; Winnie Xu; Eric Jang; Henryk Michalewski; Igor Mordatch (NeurIPS, 2022)

Chain of Thought Imitation with Procedure Cloning. - Mengjiao Yang 001; Dale Schuurmans; Pieter Abbeel; Ofir Nachum (NeurIPS, 2022)

Offline RL for Natural Language Generation with Implicit Language Q Learning. - Charlie Snell; Ilya Kostrikov; Yi Su; Sherry Yang 001; Sergey Levine (ICLR, 2023)

Dichotomy of Control: Separating What You Can Control from What You Cannot. - Sherry Yang 001; Dale Schuurmans; Pieter Abbeel; Ofir Nachum (ICLR, 2023)

Multi-Environment Pretraining Enables Transfer to Action Limited Datasets. - David Venuto; Sherry Yang 001; Pieter Abbeel; Doina Precup; Igor Mordatch; Ofir Nachum (ICML, 2023)

Learning Universal Policies via Text-Guided Video Generation. - Yilun Du; Sherry Yang 001; Bo Dai 001; Hanjun Dai; Ofir Nachum; Josh Tenenbaum 001; Dale Schuurmans; Pieter Abbeel (NeurIPS, 2023)

Foundation Models for Decision Making: Problems, Methods, and Opportunities. - Sherry Yang 001; Ofir Nachum; Yilun Du; Jason Wei; Pieter Abbeel; Dale Schuurmans (CoRR, 2023)

Probabilistic Adaptation of Text-to-Video Models. - Mengjiao Yang 001; Yilun Du; Bo Dai 001; Dale Schuurmans; Joshua B. Tenenbaum; Pieter Abbeel (CoRR, 2023)

Learning Interactive Real-World Simulators. - Mengjiao Yang 001; Yilun Du; Kamyar Ghasemipour; Jonathan Tompson; Dale Schuurmans; Pieter Abbeel (CoRR, 2023)

Video Language Planning. - Yilun Du; Sherry Yang 001; Pete Florence; Fei Xia 002; Ayzaan Wahid; Brian Ichter; Pierre Sermanet; Tianhe Yu; Pieter Abbeel; Joshua B. Tenenbaum; Leslie Pack Kaelbling; Andy Zeng 001; et al. (ICLR, 2024)

Scalable Diffusion for Materials Generation. - Sherry Yang 001; KwangHwan Cho; Amil Merchant; Pieter Abbeel; Dale Schuurmans; Igor Mordatch; Ekin Dogus Cubuk (ICLR, 2024)

Probabilistic Adaptation of Black-Box Text-to-Video Models. - Sherry Yang 001; Yilun Du; Bo Dai 001; Dale Schuurmans; Joshua B. Tenenbaum; Pieter Abbeel (ICLR, 2024)

Learning Interactive Real-World Simulators. - Sherry Yang 001; Yilun Du; Seyed Kamyar Seyed Ghasemipour; Jonathan Tompson; Leslie Pack Kaelbling; Dale Schuurmans; Pieter Abbeel (ICLR, 2024)

Code as Reward: Empowering Reinforcement Learning with VLMs. - David Venuto; Mohammad Sami Nur Islam; Martin Klissarov; Doina Precup; Sherry Yang 001; Ankit Anand (ICML, 2024)

Position: Video as the New Language for Real-World Decision Making. - Sherry Yang 001; Jacob C. Walker; Jack Parker-Holder; Yilun Du; Jake Bruce; André Barreto 001; Pieter Abbeel; Dale Schuurmans (ICML, 2024)

Generative Hierarchical Materials Search. - Sherry Yang 001; Simon L. Batzner; Ruiqi Gao; Muratahan Aykol; Alexander L. Gaunt; Brendan McMorrow; Danilo Jimenez Rezende; Dale Schuurmans; Igor Mordatch; Ekin Dogus Cubuk (NeurIPS, 2024)

UQE: A Query Engine for Unstructured Databases. - Hanjun Dai; Bethany Wang; Xingchen Wan; Bo Dai 001; Sherry Yang 001; Azade Nova; Pengcheng Yin; Phitchaya Mangpo Phothilimthana; Charles Sutton; Dale Schuurmans (NeurIPS, 2024)

Video as the New Language for Real-World Decision Making. - Sherry Yang 001; Jacob C. Walker; Jack Parker-Holder; Yilun Du; Jake Bruce; André Barreto 001; Pieter Abbeel; Dale Schuurmans (CoRR, 2024)

VideoAgent: Self-Improving Video Generation. - Achint Soni; Sreyas Venkataraman; Abhranil Chandra; Sebastian Fischmeister; Percy Liang; Bo Dai 001; Sherry Yang 001 (CoRR, 2024)

Improving Dynamic Object Interactions in Text-to-Video Generation with AI Feedback. - Hiroki Furuta; Heiga Zen; Dale Schuurmans; Aleksandra Faust; Yutaka Matsuo; Percy Liang; Sherry Yang 001 (CoRR, 2024)

Value-Incentivized Preference Optimization: A Unified Approach to Online and Offline RLHF. - Shicong Cen; Jincheng Mei; Katayoon Goshvadi; Hanjun Dai; Tong Yang 007; Sherry Yang 001; Dale Schuurmans; Yuejie Chi; Bo Dai 001 (ICLR, 2025)

System of Agentic AI for the Discovery of Metal-Organic Frameworks. - Théo Jaffrelot Inizan; Sherry Yang 001; Aaron D. Kaplan; Yen-hsu Lin; Jian Yin 030; Saber Mirzaei; Mona Abdelgaid; Ali H. Alawadhi; KwangHwan Cho; Zhiling Zheng; Ekin Dogus Cubuk; Christian Borgs; et al. (CoRR, 2025)

MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering. - Rushi Qiang; Yuchen Zhuang; Yinghao Li; Dingu Sagar V. K; Rongzhi Zhang; Changhao Li; Ian Shu-Hei Wong; Sherry Yang 001; Percy Liang; Chao Zhang 014; Bo Dai 001 (CoRR, 2025)

Evaluating Robot Policies in a World Model. - Julian Quevedo; Percy Liang; Sherry Yang 001 (CoRR, 2025)

Object-centric 3D Motion Field for Robot Learning from Human Videos. - Zhao-Heng Yin; Sherry Yang 001; Pieter Abbeel (CoRR, 2025)