elixir-nx/nx

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

解決的問題

為 Elixir 語言提供數值計算與機器學習的基礎建構模組,讓開發者能處理多維陣列(張量)並在高性能後端上執行。

工作原理

此專案由三個主要元件組成:Nx 提供張量與數值運算的定義與 API;EXLA 提供與 Google 的 XLA(加速線性代數)編譯器的整合,以加速運算;Torchx 提供基於 LibTorch 的後端。

適用對象

希望在 Elixir 生態系中進行數值計算或建構機器學習模型的 Elixir 開發者。

主要亮點

  • 支援 Elixir 的多維陣列(張量)。
  • 與 Google 的 XLA 編譯器整合,實現高效能執行。
  • 透過 Torchx 支援 LibTorch 作為後端。

相關

  • 專案
  • 專案
  • 專案
  • 專案
  • 專案