Titan: A Parallel Asynchronous Library for Multi-Agent and Soft-Body Robotics using NVIDIA CUDA. - Jacob Austin; Rafael Corrales-Fatou; Sofia Wyetzner; Hod Lipson (ICRA, 2020)

Structured Denoising Diffusion Models in Discrete State-Spaces. - Jacob Austin; Daniel D. Johnson 001; Jonathan Ho; Daniel Tarlow; Rianne van den Berg (NeurIPS, 2021)

Beyond In-Place Corruption: Insertion and Deletion In Denoising Probabilistic Models. - Daniel D. Johnson 001; Jacob Austin; Rianne van den Berg; Daniel Tarlow (CoRR, 2021)

Program Synthesis with Large Language Models. - Jacob Austin; Augustus Odena; Maxwell I. Nye; Maarten Bosma; Henryk Michalewski; David Dohan; Ellen Jiang; Carrie J. Cai; Michael Terry; Quoc V. Le; Charles Sutton (CoRR, 2021)

Show Your Work: Scratchpads for Intermediate Computation with Language Models. - Maxwell I. Nye; Anders Johan Andreassen; Guy Gur-Ari; Henryk Michalewski; Jacob Austin; David Bieber; David Dohan; Aitor Lewkowycz; Maarten Bosma; David Luan; Charles Sutton; Augustus Odena (CoRR, 2021)

Scaling Up Models and Data with t5x and seqio. - Adam Roberts; Hyung Won Chung; Anselm Levskaya; Gaurav Mishra; James Bradbury; Daniel Andor; Sharan Narang; Brian Lester; Colin Gaffney; Afroz Mohiuddin; Curtis Hawthorne; Aitor Lewkowycz; et al. (CoRR, 2022)

A Massively-Parallel 3D Simulator for Soft and Hybrid Robots. - Joel Clay; Sofia Wyetzner; Alex Gaudio; Boxi Xia; Andrew Moshova; Jacob Austin; Max Segan; Hod Lipson (CoRR, 2022)

Language Model Cascades. - David Dohan; Winnie Xu; Aitor Lewkowycz; Jacob Austin; David Bieber; Raphael Gontijo Lopes; Yuhuai Wu; Henryk Michalewski; Rif A. Saurous; Jascha Sohl-Dickstein; Kevin Murphy 002; Charles Sutton (CoRR, 2022)

Measuring the Impact of Programming Language Distribution. - Gabriel Orlanski; Kefan Xiao; Xavier Garcia; Jeffrey Hui; Joshua Howland; Jonathan Malmaud; Jacob Austin; Rishabh Singh; Michele Catasta (ICML, 2023)

PaLM 2 Technical Report. - Rohan Anil; Andrew M. Dai; Orhan Firat; Melvin Johnson; Dmitry Lepikhin; Alexandre Passos; Siamak Shakeri; Emanuel Taropa; Paige Bailey; Zhifeng Chen; Eric Chu; Jonathan H. Clark; et al. (CoRR, 2023)

PaLM: Scaling Language Modeling with Pathways. - Aakanksha Chowdhery; Sharan Narang; Jacob Devlin; Maarten Bosma; Gaurav Mishra; Adam Roberts; Paul Barham 001; Hyung Won Chung; Charles Sutton; Sebastian Gehrmann; Parker Schuh; Kensen Shi; et al. (J. Mach. Learn. Res., 2023)

Scaling Up Models and Data with t5x and seqio. - Adam Roberts; Hyung Won Chung; Gaurav Mishra; Anselm Levskaya; James Bradbury; Daniel Andor; Sharan Narang; Brian Lester; Colin Gaffney; Afroz Mohiuddin; Curtis Hawthorne; Aitor Lewkowycz; et al. (J. Mach. Learn. Res., 2023)

Resolving Code Review Comments with Machine Learning. - Alexander Frömmgen; Jacob Austin; Peter Choy; Nimesh Ghelani; Lera Kharatyan; Gabriela Surita; Elena Khrapko; Pascal Lamblin; Pierre-Antoine Manzagol; Marcus Revaj; Maxim Tabachnyk; Daniel Tarlow; et al. (ICSE-SEIP, 2024)

Gemma: Open Models Based on Gemini Research and Technology. - Thomas Mesnard; Cassidy Hardin; Robert Dadashi; Surya Bhupatiraju; Shreya Pathak; Laurent Sifre; Morgane Rivière; Mihir Sanjay Kale; Juliette Love; Pouya Tafti; Léonard Hussenot; Aakanksha Chowdhery; et al. (CoRR, 2024)