elixir-nx/nx

解决的问题

为 Elixir 语言提供数值计算和机器学习的基础构建模块，使开发者能够处理多维数组（张量）并在高性能后端上执行。

工作原理

该项目由三个主要组件组成：Nx 提供张量和数值操作的定义与 API；EXLA 提供与 Google 的 XLA（加速线性代数）编译器的集成，以加速计算；Torchx 提供基于 LibTorch 的后端。

适用人群

希望在 Elixir 生态系统中进行数值计算或构建机器学习模型的 Elixir 开发者。

主要亮点

• 支持 Elixir 的多维数组（张量）。
• 与 Google 的 XLA 编译器集成，实现高性能执行。
• 通过 Torchx 支持 LibTorch 作为后端。