publications

A complete list of my publications.

2024

  1. llemma.png
    Llemma: An Open Language Model For Mathematics
    Zhangir Azerbayev, Hailey Schoelkopf, Keiran Paster, Marco Dos Santos, Stephen McAleer, Albert Q. Jiang, Jia Deng, Stella Biderman, and Sean Welleck
    In International Conference on Learning Representations, 2024
    arXiv Code Website
  2. owm-color.png
    OpenWebMath: An Open Dataset of High-Quality Mathematical Web Text
    Keiran Paster*, Marco Dos Santos*, Zhangir Azerbayev, and Jimmy Ba
    In International Conference on Learning Representations, 2024
    arXiv Code Website

2023

  1. steve-1.gif
    STEVE-1: A Generative Model for Text-to-Behavior in Minecraft
    Shalev Lifshitz*, Keiran Paster*, Harris Chan, Jimmy Ba, and Sheila McIlraith
    In Advances in Neural Information Processing Systems, 2023
    Spotlight
    arXiv Code Website
  2. ape-algo.gif
    Large Language Models are Human-Level Prompt Engineers
    Yongchao Zhou*, Andrei Ioan Muresanu*, Ziwen Han*, Keiran Paster, Silviu Pitis, Harris Chan, and Jimmy Ba
    In International Conference on Learning Representations, 2023
    arXiv Code Website

2022

  1. esper.gif
    You Can’t Count on Luck: Why Decision Transformers and RvS Fail in Stochastic Environments
    Keiran Paster, Sheila A. McIlraith, and Jimmy Ba
    In Advances in Neural Information Processing Systems, 2022
    arXiv Code Website

2021

  1. blast.png
    BLAST: Latent Dynamics Models from Bootstrapping
    Keiran Paster*, Lev E McKinney*, Sheila A. McIlraith, and Jimmy Ba
    In Deep Reinforcement Learning Workshop NeurIPS, 2021
    PDF
  2. pa-skew.png
    Learning Domain Invariant Representations in Goal-conditioned Block MDPs
    Beining Han, Chongyi Zheng, Harris Chan, Keiran Paster, Michael R. Zhang, and Jimmy Ba
    In Advances in Neural Information Processing Systems, 2021
    arXiv
  3. glamor.gif
    Planning from Pixels using Inverse Dynamics Models
    Keiran Paster, Sheila A. McIlraith, and Jimmy Ba
    In International Conference on Learning Representations, 2021
    arXiv Code Website

2018

  1. spar.png
    Equilibrium finding via asymmetric self-play reinforcement learning
    Jie Tang*, Keiran Paster*, and Pieter Abbeel
    In Deep Reinforcement Learning Workshop NeurIPS, 2018
    PDF