jax

27604c39 - fix typo in notebook