keras-nightly 3.14.0.dev2026012304__py3-none-any.whl → 3.14.0.dev2026012404__py3-none-any.whl

keras/_tf_keras/keras/ops/__init__.py

@@ -245,6 +245,7 @@ from keras.src.ops.numpy import mod as mod
 from keras.src.ops.numpy import moveaxis as moveaxis
 from keras.src.ops.numpy import multiply as multiply
 from keras.src.ops.numpy import nan_to_num as nan_to_num
+from keras.src.ops.numpy import nanmax as nanmax
 from keras.src.ops.numpy import nanmin as nanmin
 from keras.src.ops.numpy import nansum as nansum
 from keras.src.ops.numpy import ndim as ndim

keras/_tf_keras/keras/ops/numpy/__init__.py

@@ -129,6 +129,7 @@ from keras.src.ops.numpy import mod as mod
 from keras.src.ops.numpy import moveaxis as moveaxis
 from keras.src.ops.numpy import multiply as multiply
 from keras.src.ops.numpy import nan_to_num as nan_to_num
+from keras.src.ops.numpy import nanmax as nanmax
 from keras.src.ops.numpy import nanmin as nanmin
 from keras.src.ops.numpy import nansum as nansum
 from keras.src.ops.numpy import ndim as ndim

keras/ops/__init__.py

@@ -245,6 +245,7 @@ from keras.src.ops.numpy import mod as mod
 from keras.src.ops.numpy import moveaxis as moveaxis
 from keras.src.ops.numpy import multiply as multiply
 from keras.src.ops.numpy import nan_to_num as nan_to_num
+from keras.src.ops.numpy import nanmax as nanmax
 from keras.src.ops.numpy import nanmin as nanmin
 from keras.src.ops.numpy import nansum as nansum
 from keras.src.ops.numpy import ndim as ndim

keras/ops/numpy/__init__.py

@@ -129,6 +129,7 @@ from keras.src.ops.numpy import mod as mod
 from keras.src.ops.numpy import moveaxis as moveaxis
 from keras.src.ops.numpy import multiply as multiply
 from keras.src.ops.numpy import nan_to_num as nan_to_num
+from keras.src.ops.numpy import nanmax as nanmax
 from keras.src.ops.numpy import nanmin as nanmin
 from keras.src.ops.numpy import nansum as nansum
 from keras.src.ops.numpy import ndim as ndim

keras/src/backend/jax/numpy.py

@@ -1013,6 +1013,11 @@ def moveaxis(x, source, destination):
     return jnp.moveaxis(x, source=source, destination=destination)
 
 
+def nanmax(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
+    return jnp.nanmax(x, axis=axis, keepdims=keepdims)
+
+
 def nanmin(x, axis=None, keepdims=False):
     x = convert_to_tensor(x)
     return jnp.nanmin(x, axis=axis, keepdims=keepdims)

keras/src/backend/numpy/numpy.py

@@ -960,6 +960,10 @@ def moveaxis(x, source, destination):
     return np.moveaxis(x, source=source, destination=destination)
 
 
+def nanmax(x, axis=None, keepdims=False):
+    return np.nanmax(x, axis=axis, keepdims=keepdims)
+
+
 def nanmin(x, axis=None, keepdims=False):
     return np.nanmin(x, axis=axis, keepdims=keepdims)
 

keras/src/backend/openvino/numpy.py

@@ -2107,6 +2107,10 @@ def moveaxis(x, source, destination):
     return OpenVINOKerasTensor(ov_opset.transpose(x, axes_const).output(0))
 
 
+def nanmax(x, axis=None, keepdims=False):
+    raise NotImplementedError("`nanmax` is not supported with openvino backend")
+
+
 def nanmin(x, axis=None, keepdims=False):
     raise NotImplementedError("`nanmin` is not supported with openvino backend")
 

keras/src/backend/tensorflow/numpy.py

@@ -2125,6 +2125,26 @@ def moveaxis(x, source, destination):
     return tf.transpose(x, perm)
 
 
+def nanmax(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
+
+    if not x.dtype.is_floating:
+        dtype = standardize_dtype(x.dtype)
+        if dtype == "bool":
+            return tf.reduce_any(x, axis=axis, keepdims=keepdims)
+        return tf.reduce_max(x, axis=axis, keepdims=keepdims)
+
+    x_clean = tf.where(
+        tf.math.is_nan(x), tf.constant(float("-inf"), dtype=x.dtype), x
+    )
+
+    return tf.where(
+        tf.reduce_all(tf.math.is_nan(x), axis=axis, keepdims=keepdims),
+        tf.constant(float("nan"), dtype=x.dtype),
+        tf.reduce_max(x_clean, axis=axis, keepdims=keepdims),
+    )
+
+
 def nanmin(x, axis=None, keepdims=False):
     x = convert_to_tensor(x)
 

keras/src/backend/torch/core.py

@@ -1,6 +1,7 @@
 import builtins
 import contextlib
 import functools
+import os
 
 import ml_dtypes
 import numpy as np
@@ -26,7 +27,9 @@ IS_THREAD_SAFE = True
 # Some operators such as 'aten::_foreach_mul_.Scalar'
 # are not currently implemented for the MPS device.
 # check https://github.com/pytorch/pytorch/issues/77764.
-if torch.backends.mps.is_available():
+if "KERAS_TORCH_DEVICE" in os.environ:
+    DEFAULT_DEVICE = os.environ["KERAS_TORCH_DEVICE"]
+elif torch.backends.mps.is_available():
     DEFAULT_DEVICE = "mps"
 elif torch.cuda.is_available():
     DEFAULT_DEVICE = "cuda"

keras/src/backend/torch/nn.py

@@ -1131,10 +1131,6 @@ def dot_product_attention(
     flash_attention=None,
     attn_logits_soft_cap=None,
 ):
-    if bias is not None:
-        raise ValueError(
-            "torch's `dot_product_attention` doesn't support `bias`."
-        )
     query = convert_to_tensor(query)
     key = convert_to_tensor(key)
     value = convert_to_tensor(value)
@@ -1144,6 +1140,10 @@ def dot_product_attention(
             f"Received: query.shape={query.shape}, key.shape={key.shape}, "
             f"value.shape={value.shape}."
         )
+    if bias is not None and mask is not None:
+        raise ValueError(
+            "Only one of `bias` and `mask` can be provided. Received both."
+        )
     compute_dtype = backend.result_type(query.dtype, key.dtype, value.dtype)
     query = cast(query, compute_dtype)
     key = cast(key, compute_dtype)
@@ -1154,6 +1154,9 @@ def dot_product_attention(
         # Explicit set `is_causal` to `False` when `mask` is not `None`.
         is_causal = False
         mask = torch.where(mask, 0.0, _get_large_negative(query.dtype))
+    if bias is not None:
+        bias = convert_to_tensor(bias, dtype=compute_dtype)
+        mask = bias  # Use `bias` as `mask` for scaled_dot_product_attention.
 
     axis0, axis1 = 1, 2
     query = torch.transpose(query, axis0, axis1)

keras/src/backend/torch/numpy.py

@@ -1272,6 +1272,24 @@ def moveaxis(x, source, destination):
     return torch.moveaxis(x, source=source, destination=destination)
 
 
+def nanmax(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
+    if not torch.is_floating_point(x):
+        return torch.amax(x, dim=axis, keepdim=keepdims)
+
+    if axis == () or axis == []:
+        return x
+
+    x_clean = torch.where(torch.isnan(x), float("-inf"), x)
+    out = torch.amax(x_clean, dim=axis, keepdim=keepdims)
+
+    return torch.where(
+        torch.isnan(x).all(dim=axis, keepdim=keepdims),
+        torch.tensor(float("nan"), dtype=x.dtype, device=get_device()),
+        out,
+    )
+
+
 def nanmin(x, axis=None, keepdims=False):
     x = convert_to_tensor(x)
     if not torch.is_floating_point(x):

keras/src/layers/rnn/time_distributed.py

@@ -71,37 +71,43 @@ class TimeDistributed(Wrapper):
         super().build(child_input_shape)
 
     def call(self, inputs, training=None, mask=None):
-        input_shape = ops.shape(inputs)
-        mask_shape = None if mask is None else ops.shape(mask)
-        batch_size = input_shape[0]
-        timesteps = input_shape[1]
-
-        # For TF backend with graph mode and `partial_batch_size`, skip
-        # evaluation of `batch_size` as it can be a `strided_slice` and
-        # not a constant.
-        if backend.backend() == "tensorflow":
-            from keras.src.utils.module_utils import tensorflow as tf
-
-            if (
-                not tf.executing_eagerly
-                and mask_shape is not None
-                and mask_shape[1:2] != (timesteps,)
-            ):
+        # Validate mask shape using static shape info when available
+        if mask is not None:
+            mask_shape = mask.shape
+            input_shape = inputs.shape
+
+            # Check if mask has at least 2 dimensions (batch and timesteps)
+            if len(mask_shape) < 2:
                 raise ValueError(
-                    "`TimeDistributed` Layer should be passed a `mask` of "
-                    f"shape ({batch_size}, {timesteps}, ...), "
-                    f"received: mask.shape={mask_shape}"
+                    "The `mask` passed to the `TimeDistributed` layer must be "
+                    "at least 2D (e.g., `(batch_size, timesteps)`), but it has "
+                    f"{len(mask_shape)} dimension(s) with shape {mask_shape}."
                 )
-        elif mask_shape is not None and mask_shape[:2] != (
-            batch_size,
-            timesteps,
-        ):
-            raise ValueError(
-                "`TimeDistributed` Layer should be passed a `mask` of "
-                f"shape ({batch_size}, {timesteps}, ...), "
-                f"received: mask.shape={mask_shape}"
+
+            # Check batch size and timesteps dimensions match
+            batch_mismatch = (
+                input_shape[0] is not None
+                and mask_shape[0] is not None
+                and input_shape[0] != mask_shape[0]
+            )
+            time_mismatch = (
+                input_shape[1] is not None
+                and mask_shape[1] is not None
+                and input_shape[1] != mask_shape[1]
             )
 
+            if batch_mismatch or time_mismatch:
+                raise ValueError(
+                    "The `mask` passed to the `TimeDistributed` layer has a "
+                    f"shape {mask_shape} that is incompatible with the input "
+                    f"shape {input_shape}. The first two dimensions of the "
+                    "mask (batch size and timesteps) must match the input's "
+                    "first two dimensions. Expected mask shape prefix: "
+                    f"({input_shape[0]}, {input_shape[1]})."
+                )
+
+        input_shape = ops.shape(inputs)
+
         def time_distributed_transpose(data):
             """Swaps the timestep and batch dimensions of a tensor."""
             axes = [1, 0, *range(2, len(data.shape))]
@@ -129,5 +135,7 @@ class TimeDistributed(Wrapper):
 
         # Implementation #2: use backend.vectorized_map.
 
-        outputs = backend.vectorized_map(step_function, ops.arange(timesteps))
+        outputs = backend.vectorized_map(
+            step_function, ops.arange(input_shape[0])
+        )
         return time_distributed_transpose(outputs)

keras/src/ops/core.py

@@ -277,7 +277,7 @@ def associative_scan(f, elems, reverse=False, axis=0):
     [0, 1, 3, 6, 10]
 
     >>> sum_fn = lambda x, y: [x[0] + y[0], x[1] + y[1], x[2] + y[2]]
-    >>> xs = [keras.ops.array([[1, 2]]) for _ in range(3)]
+    >>> xs = [keras.ops.array([1, 2]) for _ in range(3)]
     >>> ys = keras.ops.associative_scan(sum_fn, xs, axis=0)
     >>> ys
     [[1, 3], [1, 3], [1, 3]]

keras/src/ops/numpy.py

@@ -5064,6 +5064,63 @@ def moveaxis(x, source, destination):
     return backend.numpy.moveaxis(x, source=source, destination=destination)
 
 
+class Nanmax(Operation):
+    def __init__(self, axis=None, keepdims=False, *, name=None):
+        super().__init__(name=name)
+        self.axis = axis
+        self.keepdims = keepdims
+
+    def call(self, x):
+        return backend.numpy.nanmax(x, axis=self.axis, keepdims=self.keepdims)
+
+    def compute_output_spec(self, x):
+        dtype = dtypes.result_type(getattr(x, "dtype", backend.floatx()))
+
+        if backend.backend() == "torch" and dtype == "uint32":
+            dtype = "int32"
+
+        return KerasTensor(
+            reduce_shape(x.shape, axis=self.axis, keepdims=self.keepdims),
+            dtype=dtype,
+        )
+
+
+@keras_export(["keras.ops.nanmax", "keras.ops.numpy.nanmax"])
+def nanmax(x, axis=None, keepdims=False):
+    """Maximum of a tensor over the given axes, ignoring NaNs.
+
+    Args:
+        x: Input tensor.
+        axis: Axis or axes along which the maximum is computed.
+            The default is to compute the maximum of the flattened tensor.
+        keepdims: If this is set to `True`, the axes which are reduced are left
+            in the result as dimensions with size one. Defaults
+            to `False`.
+
+    Returns:
+        Output tensor containing the maximum, with NaN values ignored. If all
+        values along a reduced axis are NaN, the result is NaN.
+
+    Examples:
+    >>> import numpy as np
+    >>> from keras import ops
+    >>> x = np.array([[1.0, np.nan, 3.0],
+    ...               [np.nan, 2.0, 1.0]])
+    >>> ops.nanmax(x)
+    3.0
+
+    >>> ops.nanmax(x, axis=1)
+    array([3., 2.])
+
+    >>> ops.nanmax(x, axis=1, keepdims=True)
+    array([[3.],
+           [2.]])
+    """
+    if any_symbolic_tensors((x,)):
+        return Nanmax(axis=axis, keepdims=keepdims).symbolic_call(x)
+    return backend.numpy.nanmax(x, axis=axis, keepdims=keepdims)
+
+
 class Nanmin(Operation):
     def __init__(self, axis=None, keepdims=False, *, name=None):
         super().__init__(name=name)

keras/src/version.py

@@ -1,7 +1,7 @@
 from keras.src.api_export import keras_export
 
 # Unique source of truth for the version number.
-__version__ = "3.14.0.dev2026012304"
+__version__ = "3.14.0.dev2026012404"
 
 
 @keras_export("keras.version")

{keras_nightly-3.14.0.dev2026012304.dist-info → keras_nightly-3.14.0.dev2026012404.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: keras-nightly
-Version: 3.14.0.dev2026012304
+Version: 3.14.0.dev2026012404
 Summary: Multi-backend Keras
 Author-email: Keras team <keras-users@googlegroups.com>
 License: Apache License 2.0

{keras_nightly-3.14.0.dev2026012304.dist-info → keras_nightly-3.14.0.dev2026012404.dist-info}/RECORD

@@ -45,11 +45,11 @@ keras/_tf_keras/keras/losses/__init__.py,sha256=xBc_KOtSLwp3h3CKQ0EnCuIy-Bsak2SP
 keras/_tf_keras/keras/metrics/__init__.py,sha256=_wF31PTvua5ahF9JEW4Hx1UVNjVCLqVI8J5JNrZCBf8,6546
 keras/_tf_keras/keras/mixed_precision/__init__.py,sha256=AM51CzHqzcY75tqdpQiuVcTRUEpUzBqeb-EfLeSDSV8,727
 keras/_tf_keras/keras/models/__init__.py,sha256=83pyA0pzytqin8JLV6FEbPreCb-V64ToebxFGrHsVdQ,501
-keras/_tf_keras/keras/ops/__init__.py,sha256=Exo66cXhtlICdHcwCb9W1aY_kyKGmu_E0gx2g8arqAA,15667
+keras/_tf_keras/keras/ops/__init__.py,sha256=j1hmnwcCXYOBqSaKHODKVyBwwS6N38KfC-W8_t9_GmU,15716
 keras/_tf_keras/keras/ops/image/__init__.py,sha256=oM_PLh5Jk9OGfi1bbJcfWkjoq0Ye5JQG9a7v_KzDfoc,1034
 keras/_tf_keras/keras/ops/linalg/__init__.py,sha256=0ab6icK3yuIm4khSfAksGRFLEAJhaOu6gGgarau4iEQ,822
 keras/_tf_keras/keras/ops/nn/__init__.py,sha256=2eD8IlkfBrsmJjHpzsxMM3_058oGeZVgohdBd27iDnI,2992
-keras/_tf_keras/keras/ops/numpy/__init__.py,sha256=Et9qtWaBOwKAOyJJcBnUjmu40MKJz6mRHtUXHdWdH-Q,9729
+keras/_tf_keras/keras/ops/numpy/__init__.py,sha256=EEX_wcRRiw4E4Tq8Bm-n5AVi-OotP8FMdLRni12oEZQ,9778
 keras/_tf_keras/keras/optimizers/__init__.py,sha256=1fx0vEB-oGu-9dumxoIvX4qVHdgJvf74OLyYoBkE2y0,1267
 keras/_tf_keras/keras/optimizers/legacy/__init__.py,sha256=uIMQESCV80Q0FY-9ikQUjXYPyZqmTfAM3dfohQ5DzYs,516
 keras/_tf_keras/keras/optimizers/schedules/__init__.py,sha256=pQF3rQiAPuUSTUdflTr-fpL77oyGIv9xzGdjae3M3kw,1120
@@ -111,11 +111,11 @@ keras/losses/__init__.py,sha256=VIXBHQFNdLUPZ7JuwtIKj_4E-xf2yvNyrmdklvjr_xM,3667
 keras/metrics/__init__.py,sha256=qeEwtqpSCAaCr8BMUv1eVaqJl2Zb83OB5K0BG3JB0nI,6245
 keras/mixed_precision/__init__.py,sha256=AM51CzHqzcY75tqdpQiuVcTRUEpUzBqeb-EfLeSDSV8,727
 keras/models/__init__.py,sha256=83pyA0pzytqin8JLV6FEbPreCb-V64ToebxFGrHsVdQ,501
-keras/ops/__init__.py,sha256=Exo66cXhtlICdHcwCb9W1aY_kyKGmu_E0gx2g8arqAA,15667
+keras/ops/__init__.py,sha256=j1hmnwcCXYOBqSaKHODKVyBwwS6N38KfC-W8_t9_GmU,15716
 keras/ops/image/__init__.py,sha256=oM_PLh5Jk9OGfi1bbJcfWkjoq0Ye5JQG9a7v_KzDfoc,1034
 keras/ops/linalg/__init__.py,sha256=0ab6icK3yuIm4khSfAksGRFLEAJhaOu6gGgarau4iEQ,822
 keras/ops/nn/__init__.py,sha256=2eD8IlkfBrsmJjHpzsxMM3_058oGeZVgohdBd27iDnI,2992
-keras/ops/numpy/__init__.py,sha256=Et9qtWaBOwKAOyJJcBnUjmu40MKJz6mRHtUXHdWdH-Q,9729
+keras/ops/numpy/__init__.py,sha256=EEX_wcRRiw4E4Tq8Bm-n5AVi-OotP8FMdLRni12oEZQ,9778
 keras/optimizers/__init__.py,sha256=1fx0vEB-oGu-9dumxoIvX4qVHdgJvf74OLyYoBkE2y0,1267
 keras/optimizers/legacy/__init__.py,sha256=uIMQESCV80Q0FY-9ikQUjXYPyZqmTfAM3dfohQ5DzYs,516
 keras/optimizers/schedules/__init__.py,sha256=pQF3rQiAPuUSTUdflTr-fpL77oyGIv9xzGdjae3M3kw,1120
@@ -128,7 +128,7 @@ keras/regularizers/__init__.py,sha256=542Shphw7W8h4Dyf2rmqMKUECVZ8IVBvN9g1LWhz-b
 keras/saving/__init__.py,sha256=KvL2GZxjvgFgEhvEnkvqjIR9JSNHKz-NWZacXajsjLI,1298
 keras/src/__init__.py,sha256=Gi4S7EiCMkE03PbdGNpFdaUYySWDs_FcAJ8Taz9Y1BE,684
 keras/src/api_export.py,sha256=gXOkBOnmscV013WAc75lc4Up01-Kkg9EylIAT_QWctg,1173
-keras/src/version.py,sha256=zrwbXx271qaOFPQlnpEWKXzwfgQbyWOQOcBlTEB9byM,204
+keras/src/version.py,sha256=qNjRmpUKTYhBSAghCCqq1ftuxf8a-hoOqn1ML8GDtTg,204
 keras/src/activations/__init__.py,sha256=0nL3IFDB9unlrMz8ninKOWo-uCHasTUpTo1tXZb2u44,4433
 keras/src/activations/activations.py,sha256=mogPggtp4CGldI3VOPNmesRxp6EbiR1_i4KLGaVwzL8,17614
 keras/src/applications/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -171,7 +171,7 @@ keras/src/backend/jax/layer.py,sha256=o6CicT06udwamTRQIjNSDLZLyYHFzBXNbxewXgWe0i
 keras/src/backend/jax/linalg.py,sha256=LDaLZYz49ChE2kJR3YpaM9xuwusvd3krV7nNAAazTWA,2642
 keras/src/backend/jax/math.py,sha256=1IEDpdoF8e5ltu3D4wbDQuihzvJHhMXz8W9Z_E-eJqU,9391
 keras/src/backend/jax/nn.py,sha256=mrRawNvf9EWe8rdTwK_Auz6xdLkVG6hH0nIAP7hyUDE,60271
-keras/src/backend/jax/numpy.py,sha256=e-EU_q5qbWL4tQAmuXgLTzBtAReMbkJ_lHtzmP2J634,38997
+keras/src/backend/jax/numpy.py,sha256=XMBWC4MM3eOCm2tePDT1J-Bj8h6SH0VWvArGGe0-2X8,39125
 keras/src/backend/jax/optimizer.py,sha256=5DeXQHcYmUI6F9i1m1VHn3sBt4LEStOeBXnKdESevLM,4134
 keras/src/backend/jax/random.py,sha256=Uk2huGIk_dlzMrx5eDVrrr2TeCEMitn2vr4yzA0NXjs,3594
 keras/src/backend/jax/rnn.py,sha256=Ycq0qfLY4M4jhltvztpLQyywjEM17T7CZQFh4hhHOUE,7767
@@ -186,7 +186,7 @@ keras/src/backend/numpy/layer.py,sha256=dTk7W7ql7vRgll7JbOXK5PlIhQw5VHdpSjKciHd8
 keras/src/backend/numpy/linalg.py,sha256=uzLTxEyuX_gDcnoA5Q59GdTg33py0WooKK5T6T9Td6c,2543
 keras/src/backend/numpy/math.py,sha256=HdkEA5ro7dtQBTP78GFIgqTFLgNQ49PXHhqI1vLRGfo,10169
 keras/src/backend/numpy/nn.py,sha256=P9JAnTlwSTI7bVv8WIv1pDQJHpjML_WJ0RsJWy-LJMc,46200
-keras/src/backend/numpy/numpy.py,sha256=o71x6rkJgj1Mq2e0iJSIDq9v-xs_PFgskBioLc-_5cs,38168
+keras/src/backend/numpy/numpy.py,sha256=UlqAexse7-DskEkQWw5mW2mXI-vBo6NO3jAoPwMzMM4,38266
 keras/src/backend/numpy/random.py,sha256=wx2nE75q7L2cBMjtQlQx8yKMj4Ie3puFMDQsbrZO8SA,3961
 keras/src/backend/numpy/rnn.py,sha256=thOsMung1qR3lQsR4_D6hqKMFollQgrB0KwsJLk4BMY,7867
 keras/src/backend/numpy/trainer.py,sha256=MzWr8_LLHa1P6fxdUWirGw_lQwHGF_vkZ7RUGLUzjUs,11126
@@ -198,7 +198,7 @@ keras/src/backend/openvino/layer.py,sha256=5RdvaH1yOyPAphjKiuQAK1H_yZFYKE1Hp7c5b
 keras/src/backend/openvino/linalg.py,sha256=L6a4MFGND2wWzPVCh44cwuOgkcC4wJTo8Xy3HwW04lg,1614
 keras/src/backend/openvino/math.py,sha256=qw9kX2sJ2qr0dBJF12Ey0E2GcwixPUqoev6UcNra4NI,3944
 keras/src/backend/openvino/nn.py,sha256=zULPxdwVO7JDZUUtsuoEEPCLQ09ew8z8T6G_i_NEqrM,23741
-keras/src/backend/openvino/numpy.py,sha256=D1ALQlwjLfIUnWslmUbDtvmqCUF4Hy_zYoSPV_O1PIc,109841
+keras/src/backend/openvino/numpy.py,sha256=qmw4X3u0oJZptzqNHM33FPjTd_Ku6xY9dxxYFqW8c0g,109966
 keras/src/backend/openvino/random.py,sha256=4hRUtIP6qJxO3Qy9uH1x6jSuJna3nWPdUf4x2QU8-ew,5575
 keras/src/backend/openvino/rnn.py,sha256=ErmuZLPSgG9qU-NfYPPvBZ6Ysy8k-fA4g19Vhqq7OVQ,866
 keras/src/backend/openvino/trainer.py,sha256=bMmtSALqydqdS6ke-5sYW5fgxZDshDH810p_C0xCRTg,9087
@@ -211,7 +211,7 @@ keras/src/backend/tensorflow/layer.py,sha256=69d40LwL4HhKRsCjj1VRpjfrQXXF8VV3vh0
 keras/src/backend/tensorflow/linalg.py,sha256=_lZVfdY1tFvrN7xwbt3INGoTR0yC5v-kI1Q0XppVibY,8773
 keras/src/backend/tensorflow/math.py,sha256=zTu_7Ff6B2Ro862z_xH0OCmIWbV74DjsO5UnfjYuOUQ,12370
 keras/src/backend/tensorflow/nn.py,sha256=6vtZHzUED6_blUPE1Tnc3GAxPpJ2ebxoaiMn80tTL9k,51328
-keras/src/backend/tensorflow/numpy.py,sha256=j_EuTLDE8mgJSZuCt7yWHZUbvwQz3T-ZksSzCP3cl4s,105695
+keras/src/backend/tensorflow/numpy.py,sha256=NHaXiWpFYVIUrZDTz3XaUKcN1AYsI9PGnlJmksOIOY8,106315
 keras/src/backend/tensorflow/optimizer.py,sha256=kFlyEOnGjEYdLpd8mpwhUeku78__xBfZbbrDWpJrq60,9307
 keras/src/backend/tensorflow/random.py,sha256=iO8V_soaDXZm9ewyAVbjudhsMj08C348c9Bz64nxXC4,6475
 keras/src/backend/tensorflow/rnn.py,sha256=JbOSpt48cm612c7YwiTYOQCQsNXyI_6QeRhtUn8qEvM,34829
@@ -220,14 +220,14 @@ keras/src/backend/tensorflow/tensorboard.py,sha256=e7pXicuMfQjuCmq1wOmixWhWt2Ebj
 keras/src/backend/tensorflow/trackable.py,sha256=QZn0JvpBJ7Kx4e6zM2IVIWz9ADcWDB-dHN6vjoQBa9Q,1993
 keras/src/backend/tensorflow/trainer.py,sha256=otJ4CciSmE2-QH17bEyjaL-oQxiCD5eWgbkEglxjt4w,37191
 keras/src/backend/torch/__init__.py,sha256=0SiJ91WMaE_tO5q1zUsLEnU6hmPTpGKPIOkmIWaHlhk,2131
-keras/src/backend/torch/core.py,sha256=zKEwMjwSyPojYDXNBJUd2AGdjZ0KZNDf_ArYsXzJNDU,24401
+keras/src/backend/torch/core.py,sha256=cMo5dCDxWicWgezOargrq1qx-tlS1GJzzYXm--PS1l0,24506
 keras/src/backend/torch/export.py,sha256=yYc5-4JxSiaCkbFWpfCIdcm4dDBv_9uG_uH6JR6oGx0,4909
 keras/src/backend/torch/image.py,sha256=eer8LZwDMz3k2Dh5gxeTQfwrxPemM_H6eHMIP3AwRss,39149
 keras/src/backend/torch/layer.py,sha256=htECdpv9ioHWM8_zqQkEdxgDsgLu8XJi5yXgnLl-JFw,2084
 keras/src/backend/torch/linalg.py,sha256=wgPCfnscp5HOBmX9_-m-57lzxs1ttLNzmHqj2VYYq7k,2108
 keras/src/backend/torch/math.py,sha256=g-ElDii2Y_o1-t6BAu2nbS7JH-aPqVS5Fqds8aYzIlg,14324
-keras/src/backend/torch/nn.py,sha256=zmEzXEuwD7fVRDm145zsxzUDmqNmRgZS4LmeIx4Nbus,37498
-keras/src/backend/torch/numpy.py,sha256=JeMDIOubCyLqMqid1xg4CQ0Nm4gf5F7WLNggaiTviuE,58582
+keras/src/backend/torch/nn.py,sha256=80MdDzkN7wV3MJbNsGh9B8IkdBoXC36wQcV8_o13y-8,37688
+keras/src/backend/torch/numpy.py,sha256=19zGj9Yr98Aejtv_GLiqeDQjAd_MrBedpWTqGed_gYw,59093
 keras/src/backend/torch/random.py,sha256=YhLfC7qkGpzlU_i6gGPVormo3BMSo7OUA3TC3GCehrA,8292
 keras/src/backend/torch/rnn.py,sha256=MJIVbHKsUA2dZm4Gu2NvRxlrFCWeWSxSZRmFxSsC3Zg,26041
 keras/src/backend/torch/trainer.py,sha256=dcikz1c5O0FHNzRKSi6WhIHsHfLV2HDlrXPElSd1cgE,17985
@@ -459,7 +459,7 @@ keras/src/layers/rnn/lstm.py,sha256=9joVha35UQrnwQq6-khk7NbfiC9YwDvuS4kqb6S6DZ4,
 keras/src/layers/rnn/rnn.py,sha256=K4zhHjyO7vd1NvOMP3i3Bk4JWP-Ljb1C0693qmRKXA8,20171
 keras/src/layers/rnn/simple_rnn.py,sha256=35O9Yh_wQOzc5ZCVk5Zwrb15tlFFC6fMwmNxKS6v42o,17527
 keras/src/layers/rnn/stacked_rnn_cells.py,sha256=PYM7Qqp6qKFObRhu9v5NZ3iJnn-rgaCh0Hv4WebrkXY,4917
-keras/src/layers/rnn/time_distributed.py,sha256=lfmUcwmCDQXjB1hKZgQJnDU3rQ0bRwWc72SrUe7VYrg,5515
+keras/src/layers/rnn/time_distributed.py,sha256=VlSA7geAAPDB8Q5mvLD6afkcb_f7a4KbCr2ybQ0CnCA,5916
 keras/src/legacy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 keras/src/legacy/backend.py,sha256=_RMWtH3ePnZotDzOswVlgMT45rbV_bV8snJYzr-Y2AE,70002
 keras/src/legacy/layers.py,sha256=oOaFtRtroSZpKL0z4tDWOpUbsrJhmuef6twESrSOmx8,8396
@@ -496,7 +496,7 @@ keras/src/models/model.py,sha256=szseM7sjfNkdOGytF25nijhjERBu_66WPSYaJ719VBY,424
 keras/src/models/sequential.py,sha256=CC9Q1BNB9m7TkgMHRyjOzhQvneng576wJpmdgHrACKY,14352
 keras/src/models/variable_mapping.py,sha256=FVtcgjBRqOxtvkzOE6kjG9SpcB9keDg2gS5LOTlXvG0,2181
 keras/src/ops/__init__.py,sha256=aORlvnrqY_eQl0EFLWdpHsXHnQ6JLSw1qhwJMr-VXJ0,644
-keras/src/ops/core.py,sha256=t06-MvptYb6ZVwmNj083JyUtzU4M6UTVXOT2vVHtKyU,42781
+keras/src/ops/core.py,sha256=1L74Jox7wY6R_DFBzVVS3VjLlIKbE0sxyK5x-pjzx8Q,42779
 keras/src/ops/einops.py,sha256=-pxW0_AzDQNsR7t2TJrzvYXBJpmLYA3fJoO0U_U96PY,6268
 keras/src/ops/function.py,sha256=QV9n1-xeTPDK_FJ3sjlHDWVH2sqDj96R6YQnJueMOlA,17821
 keras/src/ops/image.py,sha256=Drfouun3Gaod0LNgG5nxrKkgIJ4STjWQWvzbTIjKOxs,67251
@@ -504,7 +504,7 @@ keras/src/ops/linalg.py,sha256=3V8S_cgNxZZCIFcFj-FBHTdRqWNbimDtumMvfoc0f30,26736
 keras/src/ops/math.py,sha256=4qYMJ5qAPmeSyeF63YWoGbUkQt6f4_VX0enOChU4mXU,37233
 keras/src/ops/nn.py,sha256=04gjHB2BWusy4tWm59EO5Ns1paJC5umDNGwNCKzaJWQ,104658
 keras/src/ops/node.py,sha256=aJgn9D-GkteE--Bbt2cZ9JjVxb2W2uS1OWEKoeLsl3Y,5583
-keras/src/ops/numpy.py,sha256=6-nCfjwd4y0oWLoL72ZTmyDu-kNLWdNlL4KDr6TsqC8,260893
+keras/src/ops/numpy.py,sha256=1rIBEmWGxdHUSoSjy-JRaC-GAHPCa-MZ50bfoIdUvRE,262690
 keras/src/ops/operation.py,sha256=A7sh9Hi6kZb7wkeMmhrDQIq770ofANXuP-Qg-kwCM3o,15485
 keras/src/ops/operation_utils.py,sha256=C6eThl-haKzlDH0fC1rn5-P1P-pCfIfXs-fy-ADR534,14523
 keras/src/ops/symbolic_arguments.py,sha256=MKwXxZYkyouD9BPmQ1uUNxILdcwPvTayAqXaUV3P3o4,1628
@@ -618,7 +618,7 @@ keras/utils/bounding_boxes/__init__.py,sha256=jtvQll4u8ZY0Z96HwNhP1nxWEG9FM3gI-6
 keras/utils/legacy/__init__.py,sha256=oSYZz6uS8UxSElRaaJYWJEoweJ4GAasZjnn7fNaOlog,342
 keras/visualization/__init__.py,sha256=UKWmiy6sps4SWlmQi9WX8_Z53cPpLlphz2zIeHdwJpQ,722
 keras/wrappers/__init__.py,sha256=QkS-O5K8qGS7C3sytF8MpmO6PasATpNVGF8qtb7Ojsw,407
-keras_nightly-3.14.0.dev2026012304.dist-info/METADATA,sha256=WLjmq7_YlxrYTWWRdBTiOboC4zCj0j5pKexkiZZ6C6Y,6339
-keras_nightly-3.14.0.dev2026012304.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-keras_nightly-3.14.0.dev2026012304.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.14.0.dev2026012304.dist-info/RECORD,,
+keras_nightly-3.14.0.dev2026012404.dist-info/METADATA,sha256=pKZtGqykls2YgEuMm-dRviiJEGZMo_vUAFyym60nz_0,6339
+keras_nightly-3.14.0.dev2026012404.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+keras_nightly-3.14.0.dev2026012404.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.14.0.dev2026012404.dist-info/RECORD,,