keras-nightly 3.14.0.dev2026012004__py3-none-any.whl → 3.14.0.dev2026012104__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 nansum as nansum
 from keras.src.ops.numpy import ndim as ndim
 from keras.src.ops.numpy import negative as negative
 from keras.src.ops.numpy import nextafter as nextafter

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 nansum as nansum
 from keras.src.ops.numpy import ndim as ndim
 from keras.src.ops.numpy import negative as negative
 from keras.src.ops.numpy import nextafter as nextafter

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 nansum as nansum
 from keras.src.ops.numpy import ndim as ndim
 from keras.src.ops.numpy import negative as negative
 from keras.src.ops.numpy import nextafter as nextafter

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 nansum as nansum
 from keras.src.ops.numpy import ndim as ndim
 from keras.src.ops.numpy import negative as negative
 from keras.src.ops.numpy import nextafter as nextafter

keras/src/backend/jax/numpy.py

@@ -1013,6 +1013,11 @@ def moveaxis(x, source, destination):
     return jnp.moveaxis(x, source=source, destination=destination)
 
 
+def nansum(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
+    return jnp.nansum(x, axis=axis, keepdims=keepdims)
+
+
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
     x = convert_to_tensor(x)
     return jnp.nan_to_num(x, nan=nan, posinf=posinf, neginf=neginf)

keras/src/backend/numpy/numpy.py

@@ -960,6 +960,17 @@ def moveaxis(x, source, destination):
     return np.moveaxis(x, source=source, destination=destination)
 
 
+def nansum(x, axis=None, keepdims=False):
+    axis = standardize_axis_for_numpy(axis)
+    dtype = standardize_dtype(x.dtype)
+
+    if dtype in ("bool", "int8", "int16"):
+        dtype = "int32"
+    elif dtype in ("uint8", "uint16"):
+        dtype = "uint32"
+    return np.nansum(x, axis=axis, keepdims=keepdims).astype(dtype)
+
+
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
     return np.nan_to_num(x, nan=nan, posinf=posinf, neginf=neginf)
 

keras/src/backend/openvino/numpy.py

@@ -2056,6 +2056,10 @@ def moveaxis(x, source, destination):
     return OpenVINOKerasTensor(ov_opset.transpose(x, axes_const).output(0))
 
 
+def nansum(x, axis=None, keepdims=False):
+    raise NotImplementedError("`nansum` is not supported with openvino backend")
+
+
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
     x = get_ov_output(x)
     dtype = x.get_element_type()

keras/src/backend/tensorflow/numpy.py

@@ -2125,6 +2125,22 @@ def moveaxis(x, source, destination):
     return tf.transpose(x, perm)
 
 
+def nansum(x, axis=None, keepdims=False):
+    x = convert_to_tensor(x)
+    dtype = standardize_dtype(x.dtype)
+    x_clean = tf.where(
+        tf.math.is_nan(cast(x, config.floatx())), tf.zeros((), dtype=dtype), x
+    )
+
+    if dtype in ("bool", "int8", "int16"):
+        dtype = "int32"
+    elif dtype in ("uint8", "uint16"):
+        dtype = "uint32"
+    x_clean = cast(x_clean, dtype)
+
+    return tf.reduce_sum(x_clean, axis=axis, keepdims=keepdims)
+
+
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
     x = convert_to_tensor(x)
 

keras/src/backend/torch/numpy.py

@@ -1272,6 +1272,20 @@ def moveaxis(x, source, destination):
     return torch.moveaxis(x, source=source, destination=destination)
 
 
+def nansum(x, axis=None, keepdims=False):
+    if isinstance(x, (list, tuple)):
+        x = stack(x)
+    x = convert_to_tensor(x)
+    dtype = standardize_dtype(x.dtype)
+
+    if dtype in ("bool", "uint8", "int8", "int16"):
+        dtype = "int32"
+
+    if axis == () or axis == []:
+        return cast(torch.nan_to_num(x, nan=0), dtype)
+    return cast(torch.nansum(x, dim=axis, keepdim=keepdims), dtype)
+
+
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
     x = convert_to_tensor(x)
     return torch.nan_to_num(x, nan=nan, posinf=posinf, neginf=neginf)

keras/src/layers/preprocessing/image_preprocessing/center_crop.py

@@ -183,28 +183,26 @@ class CenterCrop(BaseImagePreprocessingLayer):
 
     def transform_images(self, images, transformation=None, training=True):
         inputs = self.backend.cast(images, self.compute_dtype)
+        inputs_shape = self.backend.shape(inputs)
+
         if self.data_format == "channels_first":
-            init_height = inputs.shape[-2]
-            init_width = inputs.shape[-1]
+            init_height = inputs_shape[-2]
+            init_width = inputs_shape[-1]
         else:
-            init_height = inputs.shape[-3]
-            init_width = inputs.shape[-2]
-
-        if init_height is None or init_width is None:
-            # Dynamic size case. TODO.
-            raise ValueError(
-                "At this time, CenterCrop can only "
-                "process images with a static spatial "
-                f"shape. Received: inputs.shape={inputs.shape}"
-            )
+            init_height = inputs_shape[-3]
+            init_width = inputs_shape[-2]
 
+        # All these operations work both with ints (static sizes) and scalar
+        # tensors (dynamic sizes).
         h_diff = init_height - self.height
         w_diff = init_width - self.width
 
-        h_start = int(h_diff / 2)
-        w_start = int(w_diff / 2)
+        h_start = h_diff // 2
+        w_start = w_diff // 2
 
-        if h_diff >= 0 and w_diff >= 0:
+        if (not isinstance(h_diff, int) or h_diff >= 0) and (
+            not isinstance(w_diff, int) or w_diff >= 0
+        ):
             if len(inputs.shape) == 4:
                 if self.data_format == "channels_first":
                     return inputs[

keras/src/ops/image.py

@@ -565,6 +565,7 @@ class ExtractPatches(Operation):
         if isinstance(size, int):
             size = (size, size)
         self.size = size
+        self.is_3d = len(self.size) == 3
         if strides is None:
             strides = size
         self.strides = strides
@@ -588,11 +589,21 @@ class ExtractPatches(Operation):
         if self.data_format == "channels_last":
             channels_in = images_shape[-1]
         else:
-            channels_in = images_shape[-3]
-        if original_ndim == 3:
-            images_shape = [1] + images_shape
-        filters = self.size[0] * self.size[1] * channels_in
-        kernel_size = (self.size[0], self.size[1])
+            channels_in = images_shape[-4] if self.is_3d else images_shape[-3]
+
+        if self.is_3d:
+            # 3D patch extraction
+            if original_ndim == 4:
+                images_shape = [1] + images_shape
+            filters = self.size[0] * self.size[1] * self.size[2] * channels_in
+            kernel_size = (self.size[0], self.size[1], self.size[2])
+        else:
+            # 2D patch extraction
+            if original_ndim == 3:
+                images_shape = [1] + images_shape
+            filters = self.size[0] * self.size[1] * channels_in
+            kernel_size = (self.size[0], self.size[1])
+
         out_shape = compute_conv_output_shape(
             images_shape,
             filters,
@@ -602,10 +613,24 @@ class ExtractPatches(Operation):
             data_format=self.data_format,
             dilation_rate=self.dilation_rate,
         )
-        if original_ndim == 3:
-            out_shape = out_shape[1:]
+
+        if self.is_3d:
+            if original_ndim == 4:
+                out_shape = out_shape[1:]
+        else:
+            if original_ndim == 3:
+                out_shape = out_shape[1:]
         return KerasTensor(shape=out_shape, dtype=images.dtype)
 
+    def get_config(self):
+        return {
+            "size": self.size,
+            "strides": self.strides,
+            "dilation_rate": self.dilation_rate,
+            "padding": self.padding,
+            "data_format": self.data_format,
+        }
+
 
 @keras_export("keras.ops.image.extract_patches")
 def extract_patches(
@@ -687,26 +712,6 @@ def extract_patches(
                 f"Received: size={size} with length {len(size)}"
             )
 
-    # Determine 2D vs 3D based on size argument
-    if not isinstance(size, int) and len(size) == 3:
-        # 3D patch extraction
-        if any_symbolic_tensors((images,)):
-            return ExtractPatches3D(
-                size=size,
-                strides=strides,
-                dilation_rate=dilation_rate,
-                padding=padding,
-                data_format=data_format,
-            ).symbolic_call(images)
-        return _extract_patches_3d(
-            images,
-            size,
-            strides,
-            dilation_rate,
-            padding,
-            data_format=data_format,
-        )
-
     # 2D patch extraction (default)
     if any_symbolic_tensors((images,)):
         return ExtractPatches(
@@ -729,6 +734,23 @@ def _extract_patches(
     dilation_rate=1,
     padding="valid",
     data_format=None,
+):
+    if not isinstance(size, int) and len(size) == 3:
+        return _extract_patches_3d(
+            images, size, strides, dilation_rate, padding, data_format
+        )
+    return _extract_patches_2d(
+        images, size, strides, dilation_rate, padding, data_format
+    )
+
+
+def _extract_patches_2d(
+    images,
+    size,
+    strides=None,
+    dilation_rate=1,
+    padding="valid",
+    data_format=None,
 ):
     if isinstance(size, int):
         patch_h = patch_w = size
@@ -768,74 +790,6 @@ def _extract_patches(
     return patches
 
 
-class ExtractPatches3D(Operation):
-    def __init__(
-        self,
-        size,
-        strides=None,
-        dilation_rate=1,
-        padding="valid",
-        data_format=None,
-        *,
-        name=None,
-    ):
-        super().__init__(name=name)
-        if isinstance(size, int):
-            size = (size, size, size)
-        elif len(size) != 3:
-            raise TypeError(
-                "Invalid `size` argument. Expected an "
-                f"int or a tuple of length 3. Received: size={size}"
-            )
-        self.size = size
-        if strides is not None:
-            if isinstance(strides, int):
-                strides = (strides, strides, strides)
-            elif len(strides) != 3:
-                raise ValueError(f"Invalid `strides` argument. Got: {strides}")
-        else:
-            strides = size
-        self.strides = strides
-        self.dilation_rate = dilation_rate
-        self.padding = padding
-        self.data_format = backend.standardize_data_format(data_format)
-
-    def call(self, volumes):
-        return _extract_patches_3d(
-            volumes,
-            self.size,
-            self.strides,
-            self.dilation_rate,
-            self.padding,
-            self.data_format,
-        )
-
-    def compute_output_spec(self, volumes):
-        volumes_shape = list(volumes.shape)
-        original_ndim = len(volumes_shape)
-        strides = self.strides
-        if self.data_format == "channels_last":
-            channels_in = volumes_shape[-1]
-        else:
-            channels_in = volumes_shape[-4]
-        if original_ndim == 4:
-            volumes_shape = [1] + volumes_shape
-        filters = self.size[0] * self.size[1] * self.size[2] * channels_in
-        kernel_size = (self.size[0], self.size[1], self.size[2])
-        out_shape = compute_conv_output_shape(
-            volumes_shape,
-            filters,
-            kernel_size,
-            strides=strides,
-            padding=self.padding,
-            data_format=self.data_format,
-            dilation_rate=self.dilation_rate,
-        )
-        if original_ndim == 4:
-            out_shape = out_shape[1:]
-        return KerasTensor(shape=out_shape, dtype=volumes.dtype)
-
-
 def _extract_patches_3d(
     volumes,
     size,
@@ -935,8 +889,11 @@ def extract_patches_3d(
     >>> patches.shape
     (3, 3, 3, 81)
     """
+    # Convert int to 3-tuple for 3D
+    if isinstance(size, int):
+        size = (size, size, size)
     if any_symbolic_tensors((volumes,)):
-        return ExtractPatches3D(
+        return ExtractPatches(
             size=size,
             strides=strides,
             dilation_rate=dilation_rate,

keras/src/ops/numpy.py

@@ -5064,6 +5064,67 @@ def moveaxis(x, source, destination):
     return backend.numpy.moveaxis(x, source=source, destination=destination)
 
 
+class Nansum(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.nansum(x, axis=self.axis, keepdims=self.keepdims)
+
+    def compute_output_spec(self, x):
+        dtype = dtypes.result_type(getattr(x, "dtype", backend.floatx()))
+
+        if dtype in ("bool", "int8", "int16"):
+            dtype = "int32"
+        elif dtype in ("uint8", "uint16"):
+            dtype = "uint32"
+
+        if backend.backend() == "torch" and dtype == "uint32":
+            dtype = "int32"
+        sparse = getattr(x, "sparse", False)
+        return KerasTensor(
+            reduce_shape(x.shape, axis=self.axis, keepdims=self.keepdims),
+            dtype=dtype,
+            sparse=sparse,
+        )
+
+
+@keras_export(["keras.ops.nansum", "keras.ops.numpy.nansum"])
+def nansum(x, axis=None, keepdims=False):
+    """Sum of a tensor over the given axes, ignoring NaNs.
+
+    Args:
+        x: Input tensor.
+        axis: Axis or axes along which the sum is computed. The default is to
+            compute the sum 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.
+
+    Returns:
+        Output tensor containing the sum, with NaN values ignored.
+
+    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.nansum(x)
+    7.0
+
+    >>> ops.nansum(x, axis=1)
+    array([4., 3.])
+
+    >>> ops.nansum(x, axis=1, keepdims=True)
+    array([[4.],
+           [3.]])
+    """
+    if any_symbolic_tensors((x,)):
+        return Nansum(axis=axis, keepdims=keepdims).symbolic_call(x)
+    return backend.numpy.nansum(x, axis=axis, keepdims=keepdims)
+
+
 class NanToNum(Operation):
     def __init__(self, nan=0.0, posinf=None, neginf=None, *, name=None):
         super().__init__(name=name)

keras/src/saving/orbax_util.py

@@ -6,15 +6,39 @@ from keras.src.utils.module_utils import ocp
 
 
 def is_orbax_checkpoint(filepath):
-    """Check if the given path is an Orbax checkpoint directory."""
-    if not os.path.exists(filepath):
+    """Check if the given path is an Orbax checkpoint directory.
+
+    This function implements custom detection logic instead of relying on
+    Orbax APIs which may be unreliable in some environments.
+    """
+    if not os.path.exists(filepath) or not os.path.isdir(filepath):
         return False
 
     try:
-        return ocp.is_orbax_checkpoint(filepath)
-    except (ImportError, AttributeError):
-        # Fallback to check for orbax.checkpoint file if Orbax API not available
-        return os.path.isfile(os.path.join(filepath, "orbax.checkpoint"))
+        # List directory contents
+        contents = os.listdir(filepath)
+
+        # A set is more efficient for membership testing
+        orbax_indicators = {
+            "orbax.checkpoint",
+            "pytree.orbax-checkpoint",
+            "checkpoint_metadata",
+        }
+
+        # Fast check for standard files
+        if not orbax_indicators.isdisjoint(contents):
+            return True
+
+        # Check for step directories or temporary files in a single pass
+        return any(
+            ".orbax-checkpoint-tmp" in item
+            or (item.isdigit() and os.path.isdir(os.path.join(filepath, item)))
+            for item in contents
+        )
+
+    except (OSError, PermissionError):
+        # If we can't read the directory, assume it's not a checkpoint
+        return False
 
 
 def find_latest_orbax_checkpoint(checkpoint_dir):

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.dev2026012004"
+__version__ = "3.14.0.dev2026012104"
 
 
 @keras_export("keras.version")

{keras_nightly-3.14.0.dev2026012004.dist-info → keras_nightly-3.14.0.dev2026012104.dist-info}/METADATA

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

{keras_nightly-3.14.0.dev2026012004.dist-info → keras_nightly-3.14.0.dev2026012104.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=-ywp71gN-DpkhtlIzzXpK0CG0CDIYW7L3TRnI1Tondo,15569
+keras/_tf_keras/keras/ops/__init__.py,sha256=b95A91bWrAp3S61ui69zIwEJUMsFYVi90E5TfIX7MpE,15618
 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=mIZ_5ZA-QOALvz9jkMwOTDw8iu2NxCGqwmZyJPA3vy4,9631
+keras/_tf_keras/keras/ops/numpy/__init__.py,sha256=jiEp6-gAl22y9Qzz1HS4T3VzDBlw4VViiXti65xkeAM,9680
 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=-ywp71gN-DpkhtlIzzXpK0CG0CDIYW7L3TRnI1Tondo,15569
+keras/ops/__init__.py,sha256=b95A91bWrAp3S61ui69zIwEJUMsFYVi90E5TfIX7MpE,15618
 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=mIZ_5ZA-QOALvz9jkMwOTDw8iu2NxCGqwmZyJPA3vy4,9631
+keras/ops/numpy/__init__.py,sha256=jiEp6-gAl22y9Qzz1HS4T3VzDBlw4VViiXti65xkeAM,9680
 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=9LZy9WW3ZU2GUsZI4b8ZLDM_EODbfD60ReUUWuXtC2s,204
+keras/src/version.py,sha256=j3Ml6phnds_K6jMOGglQnTbKSNPTKDHjsbdTSMqC_v8,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=_WCCYgFddNSs_1vd_65hxHgomKR3NHOsmbvTsfcddzc,38741
+keras/src/backend/jax/numpy.py,sha256=5C-obBCsAdY288BhjtxDIccqXDDle5eaP2yt7jfeUy8,38869
 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=Qu6Ia_EasEVt4UDOmvi6mx9HmFO2eva_E8NxSMnzr5E,37743
+keras/src/backend/numpy/numpy.py,sha256=W2P2A1_Y54xR07xmR-T4ALrqJA_SliDwmoOALeI37P0,38070
 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=sSPcjNoUxP9qsIEOf3m37I8rqV0MrGPtPEijeQpp0yk,105535
+keras/src/backend/openvino/numpy.py,sha256=nZnbUk_nNaSqKaShEUwyu7Z7buGvFNxEfiNCePuYBiM,105660
 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=nIpMvr-g81I9KF74RD4AbU4e4t-0eFa9MND2Fh1u8Tk,104623
+keras/src/backend/tensorflow/numpy.py,sha256=I5S0igFo2Mq3Q0SodRyNggip9F_gwWfch6TyvVbQj_E,105076
 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
@@ -227,7 +227,7 @@ keras/src/backend/torch/layer.py,sha256=htECdpv9ioHWM8_zqQkEdxgDsgLu8XJi5yXgnLl-
 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=gvHviedkAoEaTax89wDqUrjbUSX1ndjxicHy-PLv2Nc,57668
+keras/src/backend/torch/numpy.py,sha256=zZDkUDmph1c_D0VOsSzkYjAj4TKln7laDxypVRBsZ6o,58072
 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
@@ -392,7 +392,7 @@ keras/src/layers/preprocessing/image_preprocessing/__init__.py,sha256=47DEQpj8HB
 keras/src/layers/preprocessing/image_preprocessing/aug_mix.py,sha256=spvWYUG6GcPrYZgedaE8LIwTbYE2yvPg2Hwao9UAang,11221
 keras/src/layers/preprocessing/image_preprocessing/auto_contrast.py,sha256=gY7hmXXVTO15dswR8ISf9h_gox4zDSDih2owjzb7WmE,3930
 keras/src/layers/preprocessing/image_preprocessing/base_image_preprocessing_layer.py,sha256=Ga1Wewc0Pl9uLGUp3x6dxS2j4Lh-1o7TaOtxxo9kf5o,13853
-keras/src/layers/preprocessing/image_preprocessing/center_crop.py,sha256=4v8SeerBR5GMmoFpVA0mI_LPQ-GkcI-7a0JFWAMe5VY,10033
+keras/src/layers/preprocessing/image_preprocessing/center_crop.py,sha256=Pi9GlYTo7kZbfZpfF1FUwkwi0y9v8PcQYQAurixHaeU,9965
 keras/src/layers/preprocessing/image_preprocessing/cut_mix.py,sha256=reDSKzm15J7TR5TLrx92mWE-os2H6X0jY2Pd_ra_i_E,7877
 keras/src/layers/preprocessing/image_preprocessing/equalization.py,sha256=Q6URzVSxTxcd166oNFJsVlNO3x8EUMS0plqthDwKzu4,8659
 keras/src/layers/preprocessing/image_preprocessing/max_num_bounding_box.py,sha256=BTQaWjx-bMnwtsQDQLmeohs_VQECu1WZzPmi2PkDYHs,3435
@@ -499,12 +499,12 @@ keras/src/ops/__init__.py,sha256=aORlvnrqY_eQl0EFLWdpHsXHnQ6JLSw1qhwJMr-VXJ0,644
 keras/src/ops/core.py,sha256=t06-MvptYb6ZVwmNj083JyUtzU4M6UTVXOT2vVHtKyU,42781
 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=-UBomIodLByXhxN9aSt0JUGn-N5yulXgRZ06XpXq8mM,68743
+keras/src/ops/image.py,sha256=Drfouun3Gaod0LNgG5nxrKkgIJ4STjWQWvzbTIjKOxs,67251
 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=6oaYtC0HDCPAKjn3lG4Yd9SjYKL8nRySC0IA9aXsKbg,257233
+keras/src/ops/numpy.py,sha256=c5jXbWiE5jrGh1AteL3XsSgs1wNrpNUKxmTdThpNh-0,259129
 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
@@ -546,7 +546,7 @@ keras/src/saving/__init__.py,sha256=vnrtfvnzW7Gwtxe5COhaMoEnVYB5iDe2YlqJ-DvqFIk,
 keras/src/saving/file_editor.py,sha256=tsUo9mQbMa8433tHTnOKWFhDeathYwDb0CeWcDTTTBQ,32089
 keras/src/saving/keras_saveable.py,sha256=aGIt1ajtsaamfUq18LM6ql8JEoQzi3HwzJEuwQ9bmKE,1285
 keras/src/saving/object_registration.py,sha256=OOO-7-SNfPoFkFsR_c5jzE6aSIDIlHlnMcm9IlI_Gbs,7357
-keras/src/saving/orbax_util.py,sha256=0o05YKFjiePHgeW_d5fvuUAGzymqbJTeuquUR-7uVGE,906
+keras/src/saving/orbax_util.py,sha256=ArJI9hQODUyyvzCiXt8AS3VH6E4SL0vF02-RHBk30gU,1621
 keras/src/saving/saving_api.py,sha256=PMkxXhtNNKX8GlwIsCP8-Plt19M012wNEk7i8BhxWzo,12670
 keras/src/saving/saving_lib.py,sha256=-uSXsojqzSl19FtW5FogCclvnu_nnVU3S-Si293DNq0,58723
 keras/src/saving/serialization_lib.py,sha256=yzCTm8hin__MGA2N5M5F-8Zbts5ZJVmINbrH4wEtIwI,30334
@@ -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.dev2026012004.dist-info/METADATA,sha256=xBiVNAhlJ7BwQFGxrlFQVW5YAHVpTVEGfTXxPBoLBUc,6339
-keras_nightly-3.14.0.dev2026012004.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-keras_nightly-3.14.0.dev2026012004.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.14.0.dev2026012004.dist-info/RECORD,,
+keras_nightly-3.14.0.dev2026012104.dist-info/METADATA,sha256=X7LrSSzvSkKfxvLyQxMDiMc62ql2aJbcD2AE3hH-ogc,6339
+keras_nightly-3.14.0.dev2026012104.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+keras_nightly-3.14.0.dev2026012104.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.14.0.dev2026012104.dist-info/RECORD,,