elixir-nx/nx
A multi-dimensional array (tensor) library for Elixir with XLA and PyTorch backends.

Velocity · 7d
+1.4
★ / day
Trend
→steady
star history
Nx provides multi-dimensional arrays and numerical definitions for Elixir, enabling tensor operations similar to NumPy or PyTorch. It includes EXLA, a Google XLA compiler backend for accelerated linear algebra on GPUs, and Torchx, a LibTorch backend for PyTorch interoperability. Together they provide the foundational tensor computing infrastructure for machine learning in Elixir.