2023 Learnable latent embeddings for joint behavioural and neural analysis Steffen Schneider*, Jin Hwa Lee*, and Mackenzie Weygandt Mathis Nature, 2023 HTML 2022 Deep reinforcement learning for optimization at early design stages Lorenzo Servadei, Jin Hwa Lee, José A Arjona Medina, and 4 more authors IEEE Design & Test, 2022 HTML 2021 Out-of-distribution generalization of internal models is correlated with reward Khushdeep Singh Mann, Steffen Schneider, Alberto Chiappa, and 4 more authors In Self-Supervision for Reinforcement Learning Workshop-ICLR, 2021 PDF