nx
nx copied to clipboard
Support complex numbers in Nx.LinAlg.eigh
We need to add support for complex numbers for the function and for it's grad implementation
- [ ] BinaryBackend
- [ ] EXLA
- [ ] Torchx
- [ ] grad implementation