jax

ec68d420 - Port Tridiagonal Reduction to XLA's FFI

Port Tridiagonal Reduction to XLA's FFI This CL only contains the C++ changes. Python lowering code will be added after the forward compatibility window of 3 weeks. PiperOrigin-RevId: 685679646