jax

47c9495d - Return early from eigh for small matrices. This was accidentally removed cl/577222219.