Packages

torchx

0.13.0
0.13.0 0.12.0 0.11.0 0.10.2 0.10.1 0.10.0 0.9.2 0.9.1 0.9.0 0.8.0 0.7.3 0.7.2 0.7.1 0.7.0 0.6.4 0.6.3 0.6.1 0.6.0 0.5.3 0.5.2 0.5.1 0.5.0 0.4.2 0.4.1 0.4.0 0.3.0 0.2.1 0.2.0

LibTorch bindings and backend for Nx

HexDocs

Current section

Files

Jump to
Readme
28 Versions
2 Dependencies
Dependants
Files
Activity
torchx guides backend_documentation nx_lin_alg.md
Raw

guides/backend_documentation/nx_lin_alg.md 

# Nx.LinAlg (Torchx)



On Apple MPS (`:mps`), LU, Eigh, Solve, Determinant and Cholesky fall back to

Nx (slower). QR, SVD and triangular solve run on the CPU and copy the results

back.



## `Nx.LinAlg.solve/2`



Singular matrices raise `ArgumentError` with `"can't solve for singular matrix"`.



## `Nx.LinAlg.qr/2`



`:eps` is ignored.



## `Nx.LinAlg.eigh/2`



`:max_iter` and `:eps` are ignored, except on MPS where Nx's implementation is

used.



## `Nx.LinAlg.svd/2`



Complex inputs raise. `:max_iter` is ignored.



## `Nx.LinAlg.triangular_solve/3`



`left_side: false` raises. Singular matrices raise the same error as `solve/2`.