“…Equivariant and geometric deep learning have gained traction in recent years, showing promising results in supervised learning (Cohen & Welling, 2016;Winkels & Cohen, 2018;Weiler et al, 2018;Weiler & Cesa, 2019;Worrall et al, 2017;Fuchs et al, 2020;Thomas et al, 2018), unsupervised learning (Dey et al, 2021) and reinforcement learning (van der Pol et al, 2020;Simm et al, 2020). In single agent reinforcement learning, enforcing equivariance to group symmetries has been shown to improve data efficiency, for example with MDP homomorphic networks (van der Pol et al, 2020), trajectory augmentation (Lin et al, 2020;Mavalankar, 2020), or symmetric locomotion policies (Abdolhosseini et al, 2019). Equivariant approaches enable a single agent to learn policies more efficiently within its environment by sharing weights between state-action pairs that are equivalent under a transformation.…”