Learning Generalized Gumbel-max Causal Mechanisms