jax

95e2c17b - Update test of QDWH to use stricter tolerances and test more shapes and types.

Update test of QDWH to use stricter tolerances and test more shapes and types. Get rid of comparison with scipy.linalg.polar, since its outputs are significantly less accurate than QDWH. Since the polar decomposition is unique, comparing to a less accurate implementation does not add value. PiperOrigin-RevId: 642423757