Avoid bad nightly builds caused by expiry of compat horizon for nan-safe where grads.
Avoid bad nightly builds caused by expiry of compat horizon for nan-safe where grads.

PiperOrigin-RevId: 269356570

tensorflow/probability,tensorflow/probability