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

elixir-nx / nx

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

2.66k stars 194 forks source link

Closed polvalente closed 4 months ago

polvalente commented 4 months ago

polvalente commented 4 months ago

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