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.md
Raw

guides/backend_documentation/nx.md 

# Nx (Torchx)



See also [Nx.LinAlg](backend_documentation-nx_lin_alg.html).



Torchx is a backend, not a `defn` compiler. On Apple MPS, `f64` and `c128`

are not supported and raise.



## `Nx.runtime_call/4`



Outside of a `defn`, the callback runs directly on Torchx tensors. Inside a

`defn`, you need a compiler such as EXLA or `Nx.Defn.Evaluator` — Torchx does

not provide one.



## `Nx.to_pointer/2` and `Nx.from_pointer/5`



Not supported. Both raise.