[Pallas GPU] Enable Pallas `OpsExtraTest` in 64-bit mode #61917

Job	Run time
lint_and_typecheck	58s
build with 3.10 (py 3.10 on ubuntu-20.04-16core, x64=1)	10m 11s
Documentation - test code snippets (3.10)	1m 23s
build with 3.12 (py 3.12 on ubuntu-20.04-16core, x64=0)	5m 59s
Documentation - render documentation (3.10)	6m 45s
jax2tf_test (py 3.10 on ubuntu-latest, x64=0)	1m 19s
	26m 35s

Provide feedback