Tests

Performance optimizations #9790

Job	Run time
Test the code (jax, keras-nightly)	1h 6m 20s
Test the code (openvino, keras-nightly)	4m 10s
Test the code (jax, keras-stable)	1h 9m 31s
Test the code (jax, keras-3.13)	53m 49s
Test the code (tensorflow, keras-stable)	42m 32s
Test the code (torch, keras-stable)	25m 25s
Check the code format	2m 2s
	4h 23m 49s

Provide feedback