Keras 3.9.1

rtg0795 released this 27 Mar 17:17

· 3 commits to r3.9 since this release

54bdd19

What's Changed

Fix flash attention TPU error
Fix incorrect argument in JAX flash attention.

Full Changelog: v3.9.0...v3.9.1

Assets 2