v3.1.0

版本发布时间: 2024-03-12 00:04:05

keras-team/keras最新发布版本:v3.3.2(2024-04-23 07:39:26)

New features

Add support for int8 inference. Just call model.quantize("int8") to do an in-place conversion of a bfloat16 or float32 model to an int8 model. Note that only Dense and EinsumDense layers will be converted (this covers LLMs and all Transformers in general). We may add more supported layers over time.
Add keras.config.set_backend(backend) utility to reload a different backend.
Add keras.layers.MelSpectrogram layer for turning raw audio data into Mel spectrogram representation.
Add keras.ops.custom_gradient decorator (only for JAX and TensorFlow).
Add keras.ops.image.crop_images.
Add pad_to_aspect_ratio argument to image_dataset_from_directory.
Add keras.random.binomial and keras.random.beta functions.
Enable keras.ops.einsum to run with int8 x int8 inputs and int32 output.
Add verbose argument in all dataset-creation utilities.

Fix Functional model slicing
Fix for TF XLA compilation error for SpectralNormalization
Refactor axis logic across all backends and add support for multiple axes in expand_dims and squeeze

Full Changelog: https://github.com/keras-team/keras/compare/v3.0.5...v3.1.0