← Search

Timon Willi

8 accepted papers

2026

Training AI Co-Scientists Using Rubric Rewards

ICML 2026poster

AI co-scientists are emerging as a useful tool for human researchers, with a crucial ability being proposing a research plan for a given research goal. In this work, we study how to train language models that generate better research plans by leveraging the vast corpus of existing research papers. T…

Cited by 0SourceScholar
2024

JaxMARL: Multi-Agent RL Environments and Algorithms in JAX

NeurIPS 2024poster

Benchmarks are crucial in the development of machine learning algorithms, significantly influencing reinforcement learning (RL) research through the available environments. Traditionally, RL environments run on the CPU, which limits their scalability with the computational resources typically availa…

2024

Mixtures of Experts Unlock Parameter Scaling for Deep RL

ICML 2024spotlight

The recent rapid progress in (self) supervised learning models is in large part predicted by empirical scaling laws: a model's performance scales proportionally to its size. Analogous scaling laws remain elusive for reinforcement learning domains, however, where increasing the parameter count of a m…

2024

No Regrets: Investigating and Improving Regret Approximations for Curriculum Discovery

NeurIPS 2024poster

What data or environments to use for training to improve downstream performance is a longstanding and very topical question in reinforcement learning. In particular, Unsupervised Environment Design (UED) methods have gained recent attention as their adaptive curricula promise to enable agents to be…

2022

COLA: Consistent Learning with Opponent-Learning Awareness

ICML 2022spotlight

Learning in general-sum games is unstable and frequently leads to socially undesirable (Pareto-dominated) outcomes. To mitigate this, Learning with Opponent-Learning Awareness (LOLA) introduced opponent shaping to this setting, by accounting for each agent’s influence on their opponents’ anticipated…

2022

Model-Free Opponent Shaping

ICML 2022spotlight

In general-sum games the interaction of self-interested learning agents commonly leads to collectively worst-case outcomes, such as defect-defect in the iterated prisoner’s dilemma (IPD). To overcome this, some methods, such as Learning with Opponent-Learning Awareness (LOLA), directly shape the lea…