Learn how the mind works by building it — from Bayesian inference and causal reasoning to reinforcement learning and neural networks, with hands-on probabilistic programming.
A graduate seminar bridging cognitive science, machine learning, and probabilistic programming.
Build intuition for probability from counting to conjugate models, hierarchical Bayes, and Bayesian nonparametrics.
Write probabilistic programs from Week 2 onward. Model, condition, and run inference using GenJAX on JAX — no toy examples.
Causal reasoning, generalization, reinforcement learning, and social cognition — framed as computational-level theories of the mind.
Bridge to transformers, scaling laws, RLHF, and AI alignment — see where classical models meet modern deep learning.
A free, open-source textbook on probability and probabilistic computing that grows alongside the course.
Weekly reflections, in-class exercises, and collaborative problem-solving — not just lectures.
A Narrative Introduction to Probability covers discrete & continuous probability, Bayesian learning, mixture models, and probabilistic programming with GenJAX — all with interactive notebooks.
Read the Textbook GitHub Repo12 sessions, Fridays. Topics build cumulatively; GenJAX integration is woven throughout.
Full breakdown on the syllabus.
All four assignments are completed in GenJAX.