permutation-equivariance topic

List permutation-equivariance repositories

pymarl3

125
Stars
12
Forks
Watchers

We extend pymarl2 to pymarl3, equipping the MARL algorithms with permutation invariance and permutation equivariance properties. The enhanced algorithm achieves 100% win rates on SMAC-V1 and superior...

neural-graphs

68
Stars
5
Forks
Watchers

Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).