School of Computer Science, McGill University, Montr´eal, Canada; Mila - Quebec Artficial Intelligence Institute, Montr´eal, Canada; Samsung - SAIT AI Lab, Montr´eal, Canada; DIRO, Universit´e de Montr´eal, Montr´eal, Canada
Sékou-Oumar Kaba, Arnab Kumar Mondal, Yan Zhang, Yoshua Bengio, Siamak Ravanbakhsh
Equivariance with Learned Canonicalization Functions