update logreg laplace to use bijax

[murphyk]

https://github.com/probml/pyprobml/blob/master/notebooks/book1/10/logreg_laplace_demo.ipynb

[willtryagain]

can I work on this? I am not so familiar with tfp ( i am guessing that is bijax), but I have used TensorFlow and have done courses on Probability and Statistics.

[murphyk]

Sure. You don't need to use tfp, and we don't even need bijax.
I suggest you code the laplace approximation from scratch, similar to https://github.com/probml/pyprobml/blob/master/notebooks/book1/04/laplace_approx_beta_binom_jax.ipynb
which is used to compute fig 7.3b in book2 (see below).
You just need to compute the MAP estimate using a optimizer (eg BFGS) then compute the Hessian, to get the posterior precision.

[willtryagain]

Hey, I just wanted to inform I am still working on this. The code/math is not very easy to understand for me. However, as it is also not very long so I guess I can do it.