Shi, Kensen

12 publications

ICLR 2024 ExeDec: Execution Decomposition for Compositional Generalization in Neural Program Synthesis Kensen Shi, Joey Hong, Yinlin Deng, Pengcheng Yin, Manzil Zaheer, Charles Sutton

ICML 2024 NExT: Teaching Large Language Models to Reason About Code Execution Ansong Ni, Miltiadis Allamanis, Arman Cohan, Yinlin Deng, Kensen Shi, Charles Sutton, Pengcheng Yin

ICML 2023 Can Large Language Models Reason About Program Invariants? Kexin Pei, David Bieber, Kensen Shi, Charles Sutton, Pengcheng Yin

NeurIPSW 2023 Grounding Code Generation with Input-Output Specifications Yeming Wen, Pengcheng Yin, Kensen Shi, Henryk Michalewski, Swarat Chaudhuri, Alex Polozov

NeurIPS 2023 LambdaBeam: Neural Program Search with Higher-Order Functions and Lambdas Kensen Shi, Hanjun Dai, Wen-Ding Li, Kevin Ellis, Charles A. Sutton

JMLR 2023 PaLM: Scaling Language Modeling with Pathways Aakanksha Chowdhery, Sharan Narang, Jacob Devlin, Maarten Bosma, Gaurav Mishra, Adam Roberts, Paul Barham, Hyung Won Chung, Charles Sutton, Sebastian Gehrmann, Parker Schuh, Kensen Shi, Sasha Tsvyashchenko, Joshua Maynez, Abhishek Rao, Parker Barnes, Yi Tay, Noam Shazeer, Vinodkumar Prabhakaran, Emily Reif, Nan Du, Ben Hutchinson, Reiner Pope, James Bradbury, Jacob Austin, Michael Isard, Guy Gur-Ari, Pengcheng Yin, Toju Duke, Anselm Levskaya, Sanjay Ghemawat, Sunipa Dev, Henryk Michalewski, Xavier Garcia, Vedant Misra, Kevin Robinson, Liam Fedus, Denny Zhou, Daphne Ippolito, David Luan, Hyeontaek Lim, Barret Zoph, Alexander Spiridonov, Ryan Sepassi, David Dohan, Shivani Agrawal, Mark Omernick, Andrew M. Dai, Thanumalayan Sankaranarayana Pillai, Marie Pellat, Aitor Lewkowycz, Erica Moreira, Rewon Child, Oleksandr Polozov, Katherine Lee, Zongwei Zhou, Xuezhi Wang, Brennan Saeta, Mark Diaz, Orhan Firat, Michele Catasta, Jason Wei, Kathy Meier-Hellstern, Douglas Eck, Jeff Dean, Slav Petrov, Noah Fiedel

ICLRW 2022 Compositional Generalization and Decomposition in Neural Program Synthesis Kensen Shi, Joey Hong, Manzil Zaheer, Pengcheng Yin, Charles Sutton

ICLR 2022 CrossBeam: Learning to Search in Bottom-up Program Synthesis Kensen Shi, Hanjun Dai, Kevin Ellis, Charles Sutton

ICLR 2021 BUSTLE: Bottom-up Program Synthesis Through Learning-Guided Exploration Augustus Odena, Kensen Shi, David Bieber, Rishabh Singh, Charles Sutton, Hanjun Dai

ICML 2020 Incremental Sampling Without Replacement for Sequence Models Kensen Shi, David Bieber, Charles Sutton

ICML 2020 Learning and Evaluating Contextual Embedding of Source Code Aditya Kanade, Petros Maniatis, Gogul Balakrishnan, Kensen Shi

NeurIPSW 2020 TF-Coder: Program Synthesis for Tensor Manipulations Kensen Shi, David Bieber, Rishabh Singh