Fix JAX eigh#9

refraction-ray merged 2 commits intotensorcircuit:mastertensorcircuit/tensorcircuit-ng:masterfrom

abhinavd:fix_jaxeighabhinavd/tensorcircuit-ng:fix_jaxeighCopy head branch name to clipboard