neural_compressor.jax.quantization.layers_static
================================================

.. py:module:: neural_compressor.jax.quantization.layers_static

.. autoapi-nested-parse::

   Static quantized layer implementations for JAX-backed Keras models.



Classes
-------

.. autoapisummary::

   neural_compressor.jax.quantization.layers_static.MinMaxObserver
   neural_compressor.jax.quantization.layers_static.StaticQDQLayer
   neural_compressor.jax.quantization.layers_static.QStaticDenseMixin
   neural_compressor.jax.quantization.layers_static.QStaticDense
   neural_compressor.jax.quantization.layers_static.QStaticEinsumDense
   neural_compressor.jax.quantization.layers_static.QStaticMultiHeadAttention
   neural_compressor.jax.quantization.layers_static.QStaticCachedGemma3Attention
   neural_compressor.jax.quantization.layers_static.QStaticGemma3VisionAttention
   neural_compressor.jax.quantization.layers_static.QStaticRotaryEmbedding
   neural_compressor.jax.quantization.layers_static.QStaticReversibleEmbedding


Functions
---------

.. autoapisummary::

   neural_compressor.jax.quantization.layers_static.register_static_quantized_layer


Module Contents
---------------

.. py:function:: register_static_quantized_layer(clso)

   Register quantized layer class for an original layer class.

   :param clso: Original layer class to map to a quantized implementation.
   :type clso: type

   :returns: Decorator that registers the quantized class.
   :rtype: Callable


.. py:class:: MinMaxObserver(*args, **kwargs)



   Observer that tracks running min/max values for calibration.


.. py:class:: StaticQDQLayer(name, activation_dtype, asymmetric=False, const_scale=False)



   Layer that applies static quantize-dequantize to activations.


.. py:class:: QStaticDenseMixin



   Mixin that adds static quantization to dense-like layers.


.. py:class:: QStaticDense



   Statically quantized Dense layer.


.. py:class:: QStaticEinsumDense



   Statically quantized EinsumDense layer.


.. py:class:: QStaticMultiHeadAttention



   Statically quantized MultiHeadAttention layer.


.. py:class:: QStaticCachedGemma3Attention



   Statically quantized CachedGemma3Attention layer.


.. py:class:: QStaticGemma3VisionAttention



   Statically quantized Gemma3VisionAttention layer.


.. py:class:: QStaticRotaryEmbedding



   Statically quantized RotaryEmbedding layer.


.. py:class:: QStaticReversibleEmbedding



   Statically quantized ReversibleEmbedding layer.