Settling the Variance of Multi-Agent Policy Gradients. - Jakub Grudzien Kuba; Muning Wen; Linghui Meng 001; Shangding Gu; Haifeng Zhang 002; David Mguni; Jun Wang 012; Yaodong Yang 001 (NeurIPS, 2021)

Multi-Agent Constrained Policy Optimisation. - Shangding Gu; Jakub Grudzien Kuba; Muning Wen; Ruiqing Chen; Ziyan Wang; Zheng Tian 002; Jun Wang 012; Alois C. Knoll; Yaodong Yang 001 (CoRR, 2021)

Trust Region Policy Optimisation in Multi-Agent Reinforcement Learning. - Jakub Grudzien Kuba; Ruiqing Chen; Muning Wen; Ying Wen 001; Fanglei Sun; Jun Wang 012; Yaodong Yang 001 (ICLR, 2022)

Mirror Learning: A Unifying Framework of Policy Optimisation. - Jakub Grudzien Kuba; Christian A. Schröder de Witt; Jakob N. Foerster (ICML, 2022)

Discovered Policy Optimisation. - Chris Lu 001; Jakub Grudzien Kuba; Alistair Letcher; Luke Metz; Christian Schröder de Witt; Jakob N. Foerster (NeurIPS, 2022)

Multi-Agent Reinforcement Learning is a Sequence Modeling Problem. - Muning Wen; Jakub Grudzien Kuba; Runji Lin; Weinan Zhang 001; Ying Wen 001; Jun Wang 012; Yaodong Yang 001 (NeurIPS, 2022)

Understanding Value Decomposition Algorithms in Deep Cooperative Multi-Agent Reinforcement Learning. - Zehao Dou; Jakub Grudzien Kuba; Yaodong Yang 001 (CoRR, 2022)

Heterogeneous-Agent Mirror Learning: A Continuum of Solutions to Cooperative MARL. - Jakub Grudzien Kuba; Xidong Feng; Shiyao Ding; Hao Dong 003; Jun Wang 012; Yaodong Yang 001 (CoRR, 2022)

Heterogeneous-Agent Reinforcement Learning. - Yifan Zhong; Jakub Grudzien Kuba; Siyi Hu; Jiaming Ji; Yaodong Yang 001 (CoRR, 2023)

IDQL: Implicit Q-Learning as an Actor-Critic Method with Diffusion Policies. - Philippe Hansen-Estruch; Ilya Kostrikov; Michael Janner; Jakub Grudzien Kuba; Sergey Levine (CoRR, 2023)

Safe multi-agent reinforcement learning for multi-robot control. - Shangding Gu; Jakub Grudzien Kuba; Yuanpei Chen; Yali Du 001; Long Yang 004; Alois C. Knoll; Yaodong Yang 001 (Artif. Intell., 2023)

Functional Graphical Models: Structure Enables Offline Data-Driven Optimization. - Jakub Grudzien Kuba; Masatoshi Uehara; Pieter Abbeel; Sergey Levine (CoRR, 2024)

Cliqueformer: Model-Based Optimization with Structured Transformers. - Jakub Grudzien Kuba; Pieter Abbeel; Sergey Levine (CoRR, 2024)

Heterogeneous-Agent Reinforcement Learning. - Yifan Zhong; Jakub Grudzien Kuba; Xidong Feng; Siyi Hu; Jiaming Ji; Yaodong Yang 001 (J. Mach. Learn. Res., 2024)