Center for Neural Science, New York University, New York, NY, United States of America.|Princeton Neuroscience Institute and Department of Psychology, Princeton University, Princeton, NJ, United States of America.|DeepMind, London, United Kingdom and Gatsby Computational Neuroscience Unit, University College London, United Kingdom.|Department of Psychology and Center for Brain Science, Harvard University, Cambridge, MA, United States of America.|Princeton Neuroscience Institute and Department of Psychology, Princeton University, Princeton, NJ, United States of America.