Model-Based Reinforcement Learning for Atari. - Lukasz Kaiser; Mohammad Babaeizadeh; Piotr Milos; Blazej Osinski; Roy H. Campbell; Konrad Czechowski; Dumitru Erhan; Chelsea Finn; Piotr Kozakowski; Sergey Levine; Ryan Sepassi; George Tucker; et al. (CoRR, 2019)

Uncertainty-sensitive Learning and Planning with Ensembles. - Piotr Milos; Lukasz Kucinski; Konrad Czechowski; Piotr Kozakowski; Maciej Klimek (CoRR, 2019)

RL: Generic reinforcement learning codebase in TensorFlow. - Bryan M. Li; Alexander I. Cowen-Rivers; Piotr Kozakowski; David Tao; Siddhartha Kamalakara; Nitarshan Rajkumar; Hariharan Sezhiyan; Sicong Huang 001; Aidan N. Gomez (J. Open Source Softw., 2019)

Model Based Reinforcement Learning for Atari. - Lukasz Kaiser; Mohammad Babaeizadeh; Piotr Milos; Blazej Osinski; Roy H. Campbell; Konrad Czechowski; Dumitru Erhan; Chelsea Finn; Piotr Kozakowski; Sergey Levine; Afroz Mohiuddin; Ryan Sepassi; et al. (ICLR, 2020)

Structure and Randomness in Planning and Reinforcement Learning. - Konrad Czechowski; Piotr Januszewski; Piotr Kozakowski; Lukasz Kucinski; Piotr Milos (IJCNN, 2021)

Robust and Efficient Planning using Adaptive Entropy Tree Search. - Piotr Kozakowski; Mikolaj Pacek; Piotr Milos (CoRR, 2021)

Q-Value Weighted Regression: Reinforcement Learning with Limited Data. - Piotr Kozakowski; Lukasz Kaiser; Henryk Michalewski; Afroz Mohiuddin; Katarzyna Kanska (IJCNN, 2022)

Planning and Learning using Adaptive Entropy Tree Search. - Piotr Kozakowski; Mikolaj Pacek; Piotr Milos (IJCNN, 2022)

Molecule-Edit Templates for Efficient and Accurate Retrosynthesis Prediction. - Mikolaj Sacha; Michal Sadowski; Piotr Kozakowski; Ruard van Workum; Stanislaw Jastrzebski (CoRR, 2023)

tsGT: Stochastic Time Series Modeling With Transformer. - Lukasz Kucinski; Witold Drzewakowski; Mateusz Olko; Piotr Kozakowski; Lukasz Maziarka; Marta Emilia Nowakowska; Lukasz Kaiser; Piotr Milos (CoRR, 2024)