Learning Generalized Gumbel-max Causal Mechanisms - Lorberbom, Guy; Johnson, Daniel D.; Maddison, Chris; Tarlow, Daniel; Hazan, Tamir (Advances in Neural Information Processing Systems, 2021)