All reviewers agree this is an interesting paper that should be accepted at NeurIPS. The paper tackles a challenging problem proposing a novel, elegant and insightful approach. The authors have committed to address the main concerns raised by the reviewers by including new baseline experiments, giving more details for the experimental evaluation and providing more details and clearer explanations of the model.