[mle] Update lecture with latest code and style (#572)

* update code to use jax

* fix title case

* fix typo

* fix error

* fix H

* fix suggestions

* use numpy

* fix code line length

---------

Co-authored-by: Matt McKay <mmcky@users.noreply.github.com>

Author: kp992