keras-nightly 3.12.0.dev2025090203__py3-none-any.whl → 3.12.0.dev2025090403__py3-none-any.whl

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.12.0.dev2025090203"
+__version__ = "3.12.0.dev2025090403"
 
 
 @keras_export("keras.version")

{keras_nightly-3.12.0.dev2025090203.dist-info → keras_nightly-3.12.0.dev2025090403.dist-info}/METADATA

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

{keras_nightly-3.12.0.dev2025090203.dist-info → keras_nightly-3.12.0.dev2025090403.dist-info}/RECORD

@@ -126,7 +126,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=Cin1ggWTmgri7XO7xLkow07ZNGSjmu8f-dDtkf5xHvk,204
+keras/src/version.py,sha256=7vdIM9SpPYc0tR9Sx69WYreBdg5rhdqe-QcMNCO_39U,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
@@ -165,7 +165,7 @@ keras/src/backend/jax/core.py,sha256=kd2-Q3Nm5zp7YwyO_d6ZFlqUCPKP7_hVFAUrgD_FvIU
 keras/src/backend/jax/distribution_lib.py,sha256=-pm5qtFbzTgg_Z7sxq30x4Mjdopn-6qgXFysWoNLtZo,8663
 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=QxZeeiimUulsb3j1h3ncNxIoTYdKPO89s0kP49ZwF-w,194
+keras/src/backend/jax/layer.py,sha256=o6CicT06udwamTRQIjNSDLZLyYHFzBXNbxewXgWe0iw,308
 keras/src/backend/jax/linalg.py,sha256=dtGHRYCvoVlRX0UwbDDdunA8Vp_mA3sdqoasX4P8SbQ,2532
 keras/src/backend/jax/math.py,sha256=1IEDpdoF8e5ltu3D4wbDQuihzvJHhMXz8W9Z_E-eJqU,9391
 keras/src/backend/jax/nn.py,sha256=R0a8-WB0YCl14FpRi2CQ45MFRvHCFtPTedk0Q1LfWYc,45935
@@ -477,7 +477,7 @@ keras/src/metrics/regression_metrics.py,sha256=eLacV_8CKtzA26BJDJuncUDATuL1x8O6S
 keras/src/models/__init__.py,sha256=DPbBPSfIGgsufTfJH5U5xJOeN_Ef4FMadT7KKYg3Kjg,143
 keras/src/models/cloning.py,sha256=jwVtVVVYVasFIrln2hrzJ8bR2Xhsx9wYgEqpF1AjSvE,15786
 keras/src/models/functional.py,sha256=VqCbEAmUXdrrt2X_BdC8iUQmt-GOVtfgpTMoN8l3k-s,33667
-keras/src/models/model.py,sha256=HE-F8jSY1X-Xkkb-KH3ikI6l96Vzsc7sORWgs69rx5A,36188
+keras/src/models/model.py,sha256=mqT7eOFF2BhL0VdhtsG85-hEB3k0vwSLXAYVzvgeiA4,36178
 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
@@ -513,11 +513,10 @@ keras/src/optimizers/sgd.py,sha256=_3xanWOI0s2dISxEVT7i_tehsWakQQz2y480Iwkonas,4
 keras/src/optimizers/schedules/__init__.py,sha256=vuUuHNTev8sD2-swsuq7zqyYbmaOhDyiIE6F3dGGSZU,546
 keras/src/optimizers/schedules/learning_rate_schedule.py,sha256=Sqvd0SRHLMilsBsVXuolElJR6z5tBTphlGJ99Yq8axw,35784
 keras/src/quantizers/__init__.py,sha256=gxZQYNL-XyUYV1BQgC6V9qVH2MpqI-R1wqhM6o2Sywo,1967
-keras/src/quantizers/gptq.py,sha256=rbRi2qmZsvYG2APg6Sh0TysXSeTop_Ehp9wRqZOW12U,14509
-keras/src/quantizers/gptq_config.py,sha256=X6HTcYSE2m0m7s3btRwgWmJ09KkhULtWtabANW7vI9Y,6766
-keras/src/quantizers/gptq_core.py,sha256=WrTTnPxMoAXTMS748ROI38FAYo46Djh9-zK3IFKTmLA,13587
-keras/src/quantizers/gptq_quant.py,sha256=ZTN7xQCULgczhdKOL-9KGEw9kyA-YmK5JItVPfa8r-w,5129
-keras/src/quantizers/quantizers.py,sha256=s-SXe9339RpcpPSIwRS-pi4HBqVpL7qr9bZnpLkDL_s,22413
+keras/src/quantizers/gptq.py,sha256=PZur5-9UFu8hhH2U8nfdz27awdKEV1NWf4V81OpxWvE,17457
+keras/src/quantizers/gptq_config.py,sha256=s_3ZCh4R-3E5l64JTuQFC13ZzqhpWFHy-tGZQa1CJxk,6382
+keras/src/quantizers/gptq_core.py,sha256=o-48-UwH6lbACvWTazuqgYKBYV-_4a148zQaT3dLz1w,14782
+keras/src/quantizers/quantizers.py,sha256=aljlsok9ISTLgn6QUfK9GJBKLqGshG6BZQTW62NzEus,29520
 keras/src/random/__init__.py,sha256=BmXVYPzxbhADohoLtAEEzB3cesP7YBFDsp1qc6BWWlg,420
 keras/src/random/random.py,sha256=bUADZIVDuCghwIWTk0qBxXTxUdiNGWIdsRi8QJ3ePg4,17581
 keras/src/random/seed_generator.py,sha256=hQTFxKSGCkGdBJHMmJj-cSRfqI3AhHtVxYT0tfOU6Fw,5590
@@ -597,7 +596,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.12.0.dev2025090203.dist-info/METADATA,sha256=s7mkZeqWT6lGz1lr5t2Sr7W8mFmfVhuwliOIURlu1Eg,5970
-keras_nightly-3.12.0.dev2025090203.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-keras_nightly-3.12.0.dev2025090203.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
-keras_nightly-3.12.0.dev2025090203.dist-info/RECORD,,
+keras_nightly-3.12.0.dev2025090403.dist-info/METADATA,sha256=cEhICjKAML22jqL3yo9btobI1G2M9ARh1SpNDrIAniA,5970
+keras_nightly-3.12.0.dev2025090403.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+keras_nightly-3.12.0.dev2025090403.dist-info/top_level.txt,sha256=ptcw_-QuGZ4ZDjMdwi_Z0clZm8QAqFdvzzFnDEOTs9o,6
+keras_nightly-3.12.0.dev2025090403.dist-info/RECORD,,

keras/src/quantizers/gptq_quant.py

@@ -1,133 +0,0 @@
-from keras.src import ops
-
-
-def dequantize(input_tensor, scale, zero, maxq):
-    """The core quantization function."""
-    epsilon = ops.cast(1e-8, dtype=scale.dtype)
-    scale = ops.where(ops.equal(scale, 0), epsilon, scale)
-
-    quantized_tensor = ops.divide(input_tensor, scale)
-    quantized_tensor = ops.round(quantized_tensor)
-    q = ops.add(quantized_tensor, zero)
-    q = ops.clip(q, 0, maxq)
-
-    dequantized_tensor = ops.subtract(q, zero)
-    return ops.multiply(scale, dequantized_tensor)
-
-
-class GPTQQuantization:
-    """A class that handles the quantization of weights using GPTQ method.
-
-    This class provides methods to find quantization parameters (scale and zero)
-    for a given tensor and can be used to quantize weights in a GPTQ context.
-
-    Args:
-        weight_bits: (int) The number of bits to quantize to (e.g., 4).
-        per_channel: (bool) A flag indicating whether quantization is
-            applied per-channel (`True`) or per-tensor (`False`).
-            Defaults to `False`.
-        symmetric: (bool) A flag indicating whether symmetric (`True`) or
-            asymmetric (`False`) quantization is used. Defaults to `False`.
-        group_size: (int) The size of weight groups for quantization. A
-            value of -1 indicates that grouping is not used.
-            Defaults to -1.
-    """
-
-    def __init__(
-        self, weight_bits, per_channel=True, symmetric=False, group_size=-1
-    ):
-        self.weight_bits = weight_bits
-        self.maxq = ops.cast(
-            ops.subtract(ops.power(2, weight_bits), 1), "float32"
-        )
-        self.per_channel = per_channel
-        self.symmetric = symmetric
-        self.group_size = group_size
-
-        # These are now determined later by `find_params`
-        self.scale = None
-        self.zero = None
-
-    def find_params(self, input_tensor, weight=False):
-        """Finds quantization parameters (scale and zero) for a given tensor."""
-
-        if input_tensor is None:
-            raise ValueError("Input tensor 'input_tensor' cannot be None.")
-
-        # For weights, we typically expect at least a 2D tensor.
-        if weight and len(input_tensor.shape) < 2:
-            raise ValueError(
-                f"Input weight tensor 'input_tensor' must have a rank of at "
-                f"least 2, but got rank {len(input_tensor.shape)}."
-            )
-
-        if ops.size(input_tensor) == 0:
-            raise ValueError("Input tensor 'input_tensor' cannot be empty.")
-
-        original_shape = input_tensor.shape
-
-        if self.per_channel:
-            if weight:
-                if self.group_size != -1:
-                    input_reshaped = ops.reshape(
-                        input_tensor, [-1, self.group_size]
-                    )
-                else:
-                    input_reshaped = ops.reshape(
-                        input_tensor, [original_shape[0], -1]
-                    )
-        else:  # per-tensor
-            input_reshaped = ops.reshape(input_tensor, [1, -1])
-
-        # Find min/max values
-        min_values = ops.min(input_reshaped, axis=1)
-        max_values = ops.max(input_reshaped, axis=1)
-
-        # Apply symmetric quantization logic if enabled
-        if self.symmetric:
-            max_values = ops.maximum(ops.abs(min_values), max_values)
-            min_values = ops.where(
-                ops.less(min_values, 0), ops.negative(max_values), min_values
-            )
-
-        # Ensure range is not zero to avoid division errors
-        zero_range = ops.equal(min_values, max_values)
-        min_values = ops.where(
-            zero_range, ops.subtract(min_values, 1), min_values
-        )
-        max_values = ops.where(zero_range, ops.add(max_values, 1), max_values)
-
-        # Calculate scale and zero-point
-        self.scale = ops.divide(ops.subtract(max_values, min_values), self.maxq)
-        if self.symmetric:
-            self.zero = ops.full_like(
-                self.scale, ops.divide(ops.add(self.maxq, 1), 2)
-            )
-        else:
-            self.zero = ops.round(
-                ops.divide(ops.negative(min_values), self.scale)
-            )
-
-        # Ensure scale is non-zero
-        self.scale = ops.where(ops.less_equal(self.scale, 0), 1e-8, self.scale)
-
-        if weight:
-            # Per-channel, non-grouped case: simple reshape is correct.
-            if self.per_channel and self.group_size == -1:
-                self.scale = ops.reshape(self.scale, [-1, 1])
-                self.zero = ops.reshape(self.zero, [-1, 1])
-            elif not self.per_channel:
-                num_rows = original_shape[0]
-                self.scale = ops.tile(
-                    ops.reshape(self.scale, (1, 1)), (num_rows, 1)
-                )
-                self.zero = ops.tile(
-                    ops.reshape(self.zero, (1, 1)), (num_rows, 1)
-                )
-        if self.per_channel:
-            self.scale = ops.reshape(self.scale, [-1, 1])
-            self.zero = ops.reshape(self.zero, [-1, 1])
-
-    def ready(self):
-        """Checks if the quantization parameters have been computed."""
-        return self.scale is not None and self.zero is not None