Explicitly check for vanishing density instead of just catching grad_log_prob throw