jax

c4418c10 - Update several remaining lax.linalg primitives to use registration helper functions.

Update several remaining lax.linalg primitives to use registration helper functions. In this change, we update schur, triangular_solve, tridiagonal, and tridiagonal_solve. I batched these ones since they're all pretty straightforward. PiperOrigin-RevId: 729572705