Week 5 Tutorial (2/8) Automatic differentiation and JAX Slides Notebook 1: JAX overview Notebook 2: diffmah demo diffmah model paper