Packages

C++ library enabling more ergonomic NIFs, tailored to Elixir

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

Complex is a library for types and mathematical functions for complex numbers.

Precompiled XLA binaries

Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx

Bindings to Hugging Face Tokenizers for Elixir

Optimizers for the Nx ecosystem

Create and train neural networks in Elixir

Image processing in Nx

Pre-trained and transformer Neural Network models in Axon

Digital Signal Processing extension for Nx

Safetensors implementation for Nx

Traditional machine learning on top of Nx

Elixir binding for the hnswlib library

LibTorch bindings and backend for Nx

MLX bindings and backend for Nx

Axon model rewrites to swap supported nodes for EMLX.Fast Metal shaders

IREE compiler and runtime facilities for Nx