Update TensorFlow Probability dependency to tfp-nightly.

PiperOrigin-RevId: 875794368

Author: andyl7an

.github/workflows/ci.yml

@@ -78,7 +78,7 @@ jobs:
     strategy:
       max-parallel: 8
       matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13"]
+        python-version: ["3.11", "3.12", "3.13"]
         meridian-backend: ["tensorflow", "jax"]
     concurrency:
       group: ${{ github.workflow }}-${{ github.ref }}-${{ matrix.python-version }}-${{ matrix.meridian-backend }}

CHANGELOG.md

@@ -23,6 +23,10 @@ To release a new version (e.g. from `1.0.0` -> `2.0.0`):
 
 ## [Unreleased]
 
+* Dropped support for Python 3.10. The minimum required Python version is now
+  3.11.
+* Upgraded JAX dependencies to version `0.7.2`.
+
 ## [1.5.2] - 2026-02-10
 
 * Move `schema` package as `meridian.schema`.

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[substrates-jax]==0.26.0.dev20260130",
 ]
 # MMM proto schema.
 schema = [