Create dummy training data.

Can be called inside of the `call()` method to add a scalar loss. Example: ```python class MyLayer(Layer): ...

Validate a conv input against its kernel shape. Used by `conv`, `depthwise_conv`, and `separable_conv` — all share the convention that the ke

Clone a Functional or Sequential `Model` instance. Model cloning is similar to calling a model on new inputs, except that it creates new laye

Create a distributed dataset from the original global dataset. Args: dataset: the original global dataset instance. Retu

Returns the current default dtype policy object.

Convert a Keras model to dot format. Args: model: A Keras model instance. show_shapes: whether to display shape information.

The number of dimensions of the variable.

Slice a Tensor along the given axis.

Transforms non-negative/positive integer/integers into an integer tuple. Args: value: int or iterable of ints. The value to validate and

Updates the running average of the Hessian matrix with a new batch. This method computes the Hessian matrix for a given batch of inp

Create a simple test model compatible with 2-device sharding.

DEPRECATED.

Canonicalize an axis or axes to a tuple of non-negative integers.

Convert a tensor to a NumPy array. Only called after slicing using `__getitem__`. Args: x: the tensor to convert.

Return the serializable config of the metric.

Prepares and chunks the calibration dataloader, repeating short datasets. All processing happens on the CPU. Args: tokenizer: Th

Downloads a file from a URL if it not already in the cache. By default the file at the url `origin` is downloaded to the cache_dir `~/.keras`

Checks whether NNX specific features are enabled for the JAX backend. Returns: bool: `True` if NNX backend features are enabled, `False`

Read all summaries under the logdir into a `_SummaryFile`. Args: logdir: A path to a directory that contains zero or more event fil

Map a function over leading array axes. Like Python’s builtin map, except inputs and outputs are in the form of stacked arrays. Consider usin

DEPRECATED.

Stateless version of `apply` that returns modified variables. Args: optimizer_variables: list of tensors containing the current v

Returns model metrics as a dict if the keys match with input logs. When the training / evaluation is performed with an asynchronous steps,

Builds layers and variables. Args: query_shape: Shape of the `query` tensor. value_shape: Shape of the `value` tensor

Compile the distiller with proper integration. Arguments: optimizer: Optimizer for training the student model. loss:

Determines if a given filepath indicates a remote location. This function checks if the filepath represents a known remote pattern such

Quantize a float tensor into discrete levels [0, maxq] using per-tensor/per-channel/grouped scaling. Returns `q` (same dtype as inputs/scales

Sets vocabulary (and optionally document frequency) for this layer. This method sets the vocabulary and idf weights for this layer directly,

Instantiates the EfficientNet architecture. Args: width_coefficient: float, scaling coefficient for network width. depth_coefficient:

DEPRECATED.

Computes bin boundaries from quantiles in a input dataset. Calling `adapt()` on a `Discretization` layer is an alternative to passing

Publicly accessible method for determining the current backend. Returns: String, the name of the backend Keras is currently using. One of

Clips bounding boxes to be within the image dimensions. Args: bounding_boxes: A dictionary with 'boxes' shape `(N, 4)` or `(ba