keras-nightly 3.14.0.dev2026012904__py3-none-any.whl → 3.14.0.dev2026013104__py3-none-any.whl

keras/src/backend/jax/distribution_lib.py

@@ -3,7 +3,6 @@
 import jax
 import numpy as np
 
-from keras.src.backend.common import global_state
 from keras.src.random import seed_generator
 from keras.src.utils import jax_utils
 from keras.src.utils import rng_utils
@@ -157,23 +156,6 @@ def initialize_rng():
         # Set the global seed.
         rng_utils.set_random_seed(global_seed)
 
-    # Check if the global seed generator is set and ensure it has an initialized
-    # seed.  Otherwise, reset the seed to the global seed.
-    global_seed_generator = global_state.get_global_attribute(
-        seed_generator.GLOBAL_SEED_GENERATOR
-    )
-    if global_seed_generator is not None:
-        seed = global_seed_generator.get_config()["seed"]
-        if seed is None:
-            global_state.set_global_attribute(
-                seed_generator.GLOBAL_SEED_GENERATOR,
-                seed_generator.SeedGenerator(
-                    seed=global_seed,
-                    name=global_seed_generator.name,
-                    backend=global_seed_generator.backend,
-                ),
-            )
-
 
 def initialize(job_addresses, num_processes, process_id):
     if job_addresses and "," in job_addresses:

keras/src/backend/openvino/numpy.py

@@ -2122,7 +2122,22 @@ def nanmin(x, axis=None, keepdims=False):
 
 
 def nansum(x, axis=None, keepdims=False):
-    raise NotImplementedError("`nansum` is not supported with openvino backend")
+    x = get_ov_output(x)
+    x_type = x.get_element_type()
+
+    if not x_type.is_integral() and x_type != Type.boolean:
+        nan_mask = ov_opset.is_nan(x)
+        zero = ov_opset.constant(0, x_type)
+        x = ov_opset.select(nan_mask, zero, x).output(0)
+
+    x, axis = _resolve_axis(x, axis)
+    if axis is None:
+        return OpenVINOKerasTensor(x)
+
+    x = _upcast_type_if_needed(x)
+    result = ov_opset.reduce_sum(x, axis, keepdims).output(0)
+
+    return OpenVINOKerasTensor(result)
 
 
 def nan_to_num(x, nan=0.0, posinf=None, neginf=None):
@@ -2717,7 +2732,37 @@ def round(x, decimals=0):
 
 
 def tile(x, repeats):
-    raise NotImplementedError("`tile` is not supported with openvino backend")
+    x = get_ov_output(x)
+
+    if isinstance(repeats, int):
+        repeats = [repeats]
+    repeats = get_ov_output(repeats)
+
+    if repeats.get_element_type() != Type.i64:
+        repeats = ov_opset.convert(repeats, Type.i64)
+
+    if len(repeats.get_partial_shape()) != 1:
+        repeats = ov_opset.reshape(repeats, [-1], False)
+
+    shape_x = ov_opset.shape_of(x, Type.i64)
+    rank_x = ov_opset.shape_of(shape_x, Type.i64)
+    rank_r = ov_opset.shape_of(repeats, Type.i64)
+
+    one = ov_opset.constant(1, Type.i64)
+    zero = ov_opset.constant(0, Type.i64)
+
+    pad_x = ov_opset.maximum(ov_opset.subtract(rank_r, rank_x), zero)
+    new_x_shape = ov_opset.concat(
+        [ov_opset.broadcast(one, pad_x).output(0), shape_x], 0
+    )
+    x = ov_opset.reshape(x, new_x_shape, False)
+
+    pad_r = ov_opset.maximum(ov_opset.subtract(rank_x, rank_r), zero)
+    repeats = ov_opset.concat(
+        [ov_opset.broadcast(one, pad_r).output(0), repeats], 0
+    )
+
+    return OpenVINOKerasTensor(ov_opset.tile(x, repeats).output(0))
 
 
 def trace(x, offset=0, axis1=0, axis2=1):

keras/src/export/tfsm_layer.py

@@ -2,6 +2,7 @@ from keras.src import backend
 from keras.src import layers
 from keras.src.api_export import keras_export
 from keras.src.export.saved_model import _list_variables_used_by_fns
+from keras.src.saving import serialization_lib
 from keras.src.utils.module_utils import tensorflow as tf
 
 
@@ -146,3 +147,36 @@ class TFSMLayer(layers.Layer):
             "call_training_endpoint": self.call_training_endpoint,
         }
         return {**base_config, **config}
+
+    @classmethod
+    def from_config(cls, config, custom_objects=None, safe_mode=None):
+        """Creates a TFSMLayer from its config.
+        Args:
+            config: A Python dictionary, typically the output of `get_config`.
+            custom_objects: Optional dictionary mapping names to custom objects.
+            safe_mode: Boolean, whether to disallow loading TFSMLayer.
+                When `safe_mode=True`, loading is disallowed because TFSMLayer
+                loads external SavedModels that may contain attacker-controlled
+                executable graph code. Defaults to `True`.
+        Returns:
+            A TFSMLayer instance.
+        """
+        # Follow the same pattern as Lambda layer for safe_mode handling
+        effective_safe_mode = (
+            safe_mode
+            if safe_mode is not None
+            else serialization_lib.in_safe_mode()
+        )
+
+        if effective_safe_mode is not False:
+            raise ValueError(
+                "Requested the deserialization of a `TFSMLayer`, which "
+                "loads an external SavedModel. This carries a potential risk "
+                "of arbitrary code execution and thus it is disallowed by "
+                "default. If you trust the source of the artifact, you can "
+                "override this error by passing `safe_mode=False` to the "
+                "loading function, or calling "
+                "`keras.config.enable_unsafe_deserialization()."
+            )
+
+        return cls(**config)

keras/src/optimizers/__init__.py

@@ -25,6 +25,7 @@ ALL_OBJECTS = {
     Adagrad,
     Adamax,
     Adafactor,
+    Muon,
     Nadam,
     Ftrl,
     Lion,

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

{keras_nightly-3.14.0.dev2026012904.dist-info → keras_nightly-3.14.0.dev2026013104.dist-info}/METADATA

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

{keras_nightly-3.14.0.dev2026012904.dist-info → keras_nightly-3.14.0.dev2026013104.dist-info}/RECORD

@@ -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=L6aU5qNyjfVqWFQMoyXdfGkHRVpx3Z4Axvr6FTCf4N8,204
+keras/src/version.py,sha256=czO9hPDnPlqYI5RCPx-8o5q4MY_U5sszh1qYgRP4csc,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
@@ -164,7 +164,7 @@ keras/src/backend/common/tensor_attributes.py,sha256=X5sYeGDu9YmVBIn8oX31IeE-v-b
 keras/src/backend/common/variables.py,sha256=ENq8lbwCGoIggC3Ef92Ea9G7ej4NZJnomvJZYwp8BaI,24467
 keras/src/backend/jax/__init__.py,sha256=l_HMwAZ3oAV4Etnw9RPqbvLYPPs3CZYbgaLd_qy36ps,1495
 keras/src/backend/jax/core.py,sha256=pwsVWZF63o47StwMdwm9tRQapX_EHNnMt4r8Xe37gIg,23132
-keras/src/backend/jax/distribution_lib.py,sha256=pwUsC_p_z73-5pG0_nI1tbJ9cqMqd93-ROIteEEbY7w,9258
+keras/src/backend/jax/distribution_lib.py,sha256=y5bCEApRjDKhBeobI6h9ydjZDA8mqNjT1xYW_IsHWGw,8498
 keras/src/backend/jax/export.py,sha256=jV2yKQLzYjK72vTJmdNomWPLeNS_lDTCEKzQx_5D_-E,7368
 keras/src/backend/jax/image.py,sha256=RiYIalbIaUQdDOGpDZUBk5KNsX94Xqg7iyXGATN9V58,30482
 keras/src/backend/jax/layer.py,sha256=o6CicT06udwamTRQIjNSDLZLyYHFzBXNbxewXgWe0iw,308
@@ -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=KzwfF8A1RPvmZCiZZqM0B5XELQNfyCYMAnb60Cxn2Jw,111780
+keras/src/backend/openvino/numpy.py,sha256=wEaJROQYYiz8Yw2EWt_X_BeWJrgPMrlR-LlKMIrZE8U,113136
 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
@@ -288,7 +288,7 @@ keras/src/export/onnx.py,sha256=XsVSmNmxJDQrDuUANKFo8smOSm4BBYm35bR1AW9OWFo,8404
 keras/src/export/openvino.py,sha256=C9QNCOQ8-MwoOr8ZUqQvGzWY_CHOb8yDlMIJ9NYsLyw,7626
 keras/src/export/saved_model.py,sha256=bxcsVd87MXnw3ENKu_dbUc8JzPFqjOAPbLL0U5KqG-g,28425
 keras/src/export/tf2onnx_lib.py,sha256=cvHXS84Ocjcp1cTh5SziXAzNUsZ51RqjXNhOk5IlNDs,7234
-keras/src/export/tfsm_layer.py,sha256=1OSV8sg_ftrMQjyf_RBsNNC2sihkWCKml5Yv3M3C-NA,5998
+keras/src/export/tfsm_layer.py,sha256=7qOYvUwtHluDykQtob4VgtN88EzsAgzk8MDsWolvB38,7545
 keras/src/initializers/__init__.py,sha256=tG7qxC2J0PDhO_L2W95sJXNIduL7F5lqHvUuJ7EIhXE,5662
 keras/src/initializers/constant_initializers.py,sha256=CvTyqbkcvvhwLlKYf8jqwlS-F2-Uj2c13si8Wjc4tmQ,10072
 keras/src/initializers/initializer.py,sha256=kNAyRA8CzBdtknT6ZUt5XIO2_Z9NzpN119CId7wT1Vg,2632
@@ -508,7 +508,7 @@ keras/src/ops/numpy.py,sha256=VmjIuyjZYDu37AO0PaJ6fpw8MnVKcFUJ5IDx_nnPxps,264374
 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
-keras/src/optimizers/__init__.py,sha256=k7AmJUexCuGHTvU5gCrL_Pf7XYQmA6dZjJ47kcLvqfk,3974
+keras/src/optimizers/__init__.py,sha256=9NzqTWeXY5NwY9Z3htYsZAVmEQFFudX1eGrv8uJizX0,3984
 keras/src/optimizers/adadelta.py,sha256=QRVnzP2vdaEX02yb81Wv9ovV5VNvjBGZz3iKISBp150,4581
 keras/src/optimizers/adafactor.py,sha256=wz90cSLBM-pCtYHSnxFSSYXEo-CBa4POCmL98HWhwcs,8418
 keras/src/optimizers/adagrad.py,sha256=x6HxwrqAjyhro1HQabcJQLLPPaXhWFJH6n-K0jBvE9g,3722
@@ -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.dev2026012904.dist-info/METADATA,sha256=0pSnPO1OPi8catYTDRABuendG3SeP4dMURreOSKYJRQ,6339
-keras_nightly-3.14.0.dev2026012904.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-keras_nightly-3.14.0.dev2026012904.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.14.0.dev2026012904.dist-info/RECORD,,
+keras_nightly-3.14.0.dev2026013104.dist-info/METADATA,sha256=w0kkqL1AF73EhiimNKldJ4tSW7eCF5cKZwXLOhxviSM,6339
+keras_nightly-3.14.0.dev2026013104.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+keras_nightly-3.14.0.dev2026013104.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.14.0.dev2026013104.dist-info/RECORD,,