Tests

Fix mixed precision crash in TopP and Random samplers #9766

Job	Run time
Check the code format	2m 19s
Test the code (jax, keras-stable)	1h 4m 38s
Test the code (jax, keras-3.13)	54m 59s
Test the code (torch, keras-stable)	22m 36s
Test the code (tensorflow, keras-stable)	42m 11s
Test the code (openvino, keras-nightly)	3m 52s
Test the code (jax, keras-nightly)	1h 1m 24s
	4h 11m 59s

Provide feedback