jax

39daf07d - Add trivial implementations of eigvals/eigvalsh (#1604)

Add trivial implementations of eigvals/eigvalsh (#1604) * Add trivial implementations of eigvals/eigvalsh The implementations simply delegate to eig/eigh. * Enable eigvalsh test on TPU/GPU