[22] Tom Hennigan et al. Haiku: Sonnet for JAX. Version 0.0.9. 2020. URL: http://github.com/deepmind/dm-haiku.
[23] Irina Higgins et al. "Beta-VAE: Learning basic visual concepts with a constrained variational framework". In: International conference on learning representations. 2017.
[24] Irina Higgins et al. "Darla: Improving zero-shot transfer in reinforcement learning". In: International Conference on Machine Learning. PMLR. 2017, pp. 1480-1490.
[25] Sepp Hochreiter and Jürgen Schmidhuber. “Long short-term memory”. In: Neural computation 9.8 (1997), pp. 1735–1780.
[26] Makoto Ito and Kenji Doya. "Validation of decision-making models and analysis of decision variables in the rat basal ganglia". In: Journal of Neuroscience 29.31 (2009), pp. 9861-9874.
[27] MI Jordan. Serial order: a parallel distributed processing approach. technical report, june 1985-march 1986. Tech. rep. California Univ., San Diego, La Jolla (USA). Inst. for Cognitive Science, 1986.
[28] Hoseok Kim et al. "Role of striatum in updating values of chosen actions". In: Journal of neuroscience 29.47 (2009), pp. 14701-14712.
[29] Timothy D Kim et al. "Inferring latent dynamics underlying neural population activity via neural differential equations". In: International Conference on Machine Learning. PMLR. 2021, pp. 5551-5561.
[30] Diederik P Kingma and Jimmy Ba. "Adam: A method for stochastic optimization". In: arXiv preprint arXiv:1412.6980 (2014).
[31] Diederik P Kingma and Max Welling. "Auto-encoding variational bayes". In: arXiv preprint arXiv:1312.6114 (2013).
[32] Brian Lau and Paul W Glimcher. "Dynamic response-by-response models of matching behavior in rhesus monkeys". In: Journal of the experimental analysis of behavior 84.3 (2005), pp. 555-579.
[33] Daeyeol Lee, Benjamin P McGreevy, and Dominic J Barraclough. "Learning and decision making in monkeys during a rock-paper-scissors game". In: Cognitive Brain Research 25.2 (2005), pp. 416-430.
[34] Germain Lefebvre et al. “Behavioural and neural characterization of optimistic reinforcement learning”. In: Nature Human Behaviour 1.4 (2017), p. 0067.
[35] Niru Maheswaranathan et al. "Universality and individuality in neural dynamics across large populations of recurrent networks". In: Advances in neural information processing systems 32 (2019).
[36] Vladimir Mikulik et al. "Meta-trained agents implement bayes-optimal agents". In: Advances in neural information processing systems 33 (2020), pp. 18691-18703.
[37] Kevin J Miller, Matthew M Botvinick, and Carlos D Brody. "From predictive models to cognitive models: Separable behavioral processes underlying reward learning in the rat". In: bioRxiv (2018).
[38] Kevin J Miller, Matthew M Botvinick, and Carlos D Brody. “Value representations in the rodent orbitofrontal cortex drive learning, not choice”. In: *Elife* 11 (2022), e64575.
[39] John P O'Doherty, Alan Hampton, and Hackjin Kim. "Model-based fMRI and its application to reward learning and decision making". In: Annals of the New York Academy of sciences 1104.1 (2007), pp. 35-53.
[40] Chethan Pandarinath et al. "Inferring single-trial neural population dynamics using sequential auto-encoders". In: Nature methods 15.10 (2018), pp. 805–815.
[41] Lucas Pinto et al. "An accumulation-of-evidence task using visual pulses for mice navigating in virtual reality". In: Frontiers in behavioral neuroscience 12 (2018), p. 36.
[42] Payam Piray and Nathaniel D Daw. "A model for learning based on the joint estimation of stochasticity and volatility". In: Nature communications 12.1 (2021), p. 6587.
[43] Kazuyuki Samejima et al. “Representation of action-specific reward values in the striatum”. In: Science 310.5752 (2005), pp. 1337–1340.
[44] Steffen Schneider, Jin Hwa Lee, and Mackenzie Weygandt Mathis. "Learnable latent embeddings for joint behavioural and neural analysis". In: Nature (2023), pp. 1-9.
[45] Benjamin B Scott et al. "Sources of noise during accumulation of evidence in unrestrained and voluntarily head-restrained rats". In: *Elife* 4 (2015), e11308.