feat(exla): add custom callback for Nx.LinAlg.eigh

elixir-nx / nx

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

2.55k stars 190 forks source link

Closed polvalente closed 4 days ago

polvalente commented 4 days ago

polvalente commented 4 days ago

Merging because the failing tests are also failing in main :)