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

@@ -34,8 +34,8 @@ dependencies = [
   "pandas >= 2.2.2, < 3",
   "scipy >= 1.13.1, < 2",
   "statsmodels >= 0.14.5",
-  "tensorflow >= 2.18, < 2.21",
-  "tensorflow-probability >= 0.25, < 0.26",
+  "tensorflow >= 2.19.0, < 2.21",
+  "tfp-nightly == 0.26.0.dev20260130",
   "tf-keras >= 2.18, < 2.21",
   "xarray",
 ]
@@ -66,17 +66,17 @@ colab = [
 # GPU deps
 # Installed through `pip install -e .[and-cuda]`
 and-cuda = [
-  "tensorflow[and-cuda] >= 2.18, < 2.21",
+  "tensorflow[and-cuda] >= 2.19.0, < 2.21",
 ]
 # MLflow deps
 # Installed through `pip install -e .[mlflow]`
 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.7.2",
+    "jaxlib==0.7.2",
+    "tfp-nightly[jax]==0.26.0.dev20260130",
 ]
 # MMM proto schema.
 schema = [