Try fixing testSimplexAffine.

Before, the vast majority of the tested coordinates produced zero gradients,
causing enormous logspam from Eigen's cholesky decomposition. Reduce the range
of the unconstrained coordinates to more finely grid areas with non-zero
gradients, reducing/eliminating the logspam and hopefully increasing the
accuracy of the integration thus making the test less flaky.

PiperOrigin-RevId: 546352366

Author: SiegeLordEx

tensorflow_probability/python/experimental/tangent_spaces/simplex_test.py

@@ -36,7 +36,7 @@ def generate_coords(self):
     # Ensure that this represents the positive orthant of the sphere.
     logits = []
     for _ in range(self.dims):
-      logits.append(tf.range(-50., 50., self.delta))
+      logits.append(tf.range(-7., 7., self.delta))
     return tf.reshape(
         tf.stack(tf.meshgrid(*logits), axis=-1), [-1, self.dims])