Update TensorFlow Probability dependency to tfp-nightly.

PiperOrigin-RevId: 875794368

Author: andyl7an

meridian/schema/processors/reach_frequency_optimization_processor.py

@@ -447,7 +447,7 @@ def _create_frequency_outcome_grid(
         aggregated_reach = backend.einsum(
             f"{input_tensor_dims}->...{output_tensor_dims}", filtered_reach
         )
-        reach = aggregated_reach.numpy()[-1]
+        reach = np.asarray(aggregated_reach)[-1]
 
         metric_data_array = optimal_frequency_dataset[constants.ROI].sel(
             frequency=frequency, rf_channel=channel

pyproject.toml

@@ -35,7 +35,7 @@ dependencies = [
   "scipy >= 1.13.1, < 2",
   "statsmodels >= 0.14.5",
   "tensorflow >= 2.18, < 2.21",
-  "tensorflow-probability >= 0.25, < 0.26",
+  "tfp-nightly == 0.26.0.dev20260130",
   "tf-keras >= 2.18, < 2.21",
   "xarray",
 ]
@@ -74,9 +74,9 @@ mlflow = ["mlflow"]
 # JAX backend dependencies.
 # Installed through `pip install -e .[jax]`
 jax = [
-    "jax==0.5.3",
-    "jaxlib==0.5.3",
-    "tensorflow-probability[substrates-jax]==0.25.0",
+    "jax >= 0.6.2, <= 0.7.2",
+    "jaxlib >= 0.6.2, <= 0.7.2",
+    "tfp-nightly[substrates-jax]==0.26.0.dev20260130",
 ]
 # MMM proto schema.
 schema = [