v3.13.1
Fixed 3
- Remove a persistent warning triggered during import keras when using NumPy 2.0 or higher
- Fix an issue where CUDNN flash attention was broken when using JAX versions greater than 0.6.2
- Resolve a regression in the export pipeline that incorrectly forced batch sizes to be dynamic, now correctly respecting static batch sizes when defined
Bug Fixes & Improvements
- General
- Removed a persistent warning triggered during
import keraswhen using NumPy 2.0 or higher. (#21949)
- Removed a persistent warning triggered during
- Backends
- JAX: Fixed an issue where CUDNN flash attention was broken when using JAX versions greater than 0.6.2. (#21970)
- Export & Serialization
- Resolved a regression in the export pipeline that incorrectly forced batch sizes to be dynamic. The export process now correctly respects static batch sizes when defined. (#21944)
Full Changelog: https://github.com/keras-team/keras/compare/v3.13.0...v3.13.1