elixir-nx/nx

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

2,905

stars

2,329

commits

Elixir

primary language

Sep 10, 2026

updated

elixir

gpu

jit

numerical

pytorch

tensor

xla

README

This repository currently holds the following projects:

Nx - Multi-dimensional arrays (tensors) and numerical definitions for Elixir
EXLA - Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx
Torchx - LibTorch backend for Nx

Each has their own README, which you can access above to learn more. They will be extracted to their own repository in the future. Examples and benchmarks are available in the EXLA project.

Check our organization page for a general introduction to Machine Learning in Elixir.

Contributors

(top 30 of 113)

991 commits

499 commits

420 commits

80 commits

Similar repos

tenstorrent/tt-xla

Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT…

tracel-ai/burn

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on…

15,924

burn-rs/burn

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on…

15,924

ml-explore/mlx

MLX: An array framework for Apple silicon

28,426

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

7,319

patrick-kidger/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.…

1,869

tinygrad/tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

33,601

PrismML-Eng/mlx

MLX: An array framework for Apple silicon

Languages

Elixir

93.8%

C++

5.8%

elixir-nx/nx

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

2,905

stars

2,329

commits

Elixir

primary language

Sep 10, 2026

updated

elixir

gpu

jit

numerical

pytorch

tensor

xla

README

This repository currently holds the following projects:

Nx - Multi-dimensional arrays (tensors) and numerical definitions for Elixir
EXLA - Google's XLA (Accelerated Linear Algebra) compiler/backend for Nx
Torchx - LibTorch backend for Nx

Each has their own README, which you can access above to learn more. They will be extracted to their own repository in the future. Examples and benchmarks are available in the EXLA project.

Check our organization page for a general introduction to Machine Learning in Elixir.

Similar repos

tenstorrent/tt-xla

Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT…

tracel-ai/burn

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on…

15,924

burn-rs/burn

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on…

15,924

ml-explore/mlx

MLX: An array framework for Apple silicon

28,426

google/flax

Flax is a neural network library for JAX that is designed for flexibility.

7,319

patrick-kidger/jaxtyping

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays.…

1,869

tinygrad/tinygrad

You like pytorch? You like micrograd? You love tinygrad! ❤️

33,601

PrismML-Eng/mlx

MLX: An array framework for Apple silicon

Contributors

(top 30 of 113)

991 commits

499 commits

420 commits

80 commits

Languages

Elixir

93.8%

C++

5.8%