jax

1d12a993 - Port GPU kernel for symmetric eigendecomposition to GPU.

Port GPU kernel for symmetric eigendecomposition to GPU. Of note, I moved the logic about which algorithm to use, and when to use the batched algorithm into the kernel in order to support shape polymorphism and export. PiperOrigin-RevId: 671853879