internal

PiperOrigin-RevId: 380869895

Author: lamberta

tensorflow_probability/python/experimental/distributions/marginal_fns.py

@@ -25,9 +25,8 @@
 def make_backoff_choleksy(alternate_cholesky, name='BackoffCholesky'):
   """Make a function that tries Cholesky then the user-specified function.
 
-  Note this will NOT work under a gradient tape until b/177365178 is resolved.
-  Also this uses XLA compilation, which is necessary until b/144845034 is
-  resolved.
+  Warning: This function uses an XLA-compiled `tf.linalg.cholesky` to capture
+  factorization failures.
 
   Args:
     alternate_cholesky: A callable with the same signature as