Toy example of Logistic Regression with tensorflow probability and the titanic dataset fails

Question

I am learning tensorflow-probability and this is a toy example of logistic regression with the titanic dataset. My model does not seem to learn and the loss is nan. I don\