elixir-nx/nx

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

何を解決するか

Elixir言語における数値計算および機械学習の基盤となる構成要素を提供し、多次元配列（テンソル）を扱い、高性能バックエンド上で実行できるようにします。

動作方法

このプロジェクトは3つの主要なコンポーネントから構成されています：Nxはテンソルおよび数値演算の定義とAPIを提供します。EXLAはGoogleのXLA（Accelerated Linear Algebra）コンパイラとの統合を提供し、計算を高速化します。TorchxはLibTorchに基づくバックエンドを提供します。

対象ユーザー

Elixirエコシステム内で数値計算や機械学習モデルの構築をしたいElixir開発者です。

特徴

  • Elixir向けの多次元配列（テンソル）サポート。
  • 高性能実行のためのGoogleのXLAコンパイラとの統合。
  • Torchxを介したLibTorchバックエンドのサポート。

関連

  • プロジェクト
    elixir-nx/scholar

    Nx に基づく、Elixir 用の伝統的な機械学習アルゴリズムのライブラリです。分類、回帰、クラスタリング、前処理のツールを提供します。

  • プロジェクト
    elixir-nx/axon

    Nxによる数値高速化を備えたElixir向けニューラルネットワークフレームワーク。モデル作成と学習のための高レベルAPIを提供。

  • プロジェクト
    openxla/xla

    XLAは、PyTorch、TensorFlow、JAXのモデルをGPU、CPU、およびMLアクセラレータ上で高性能に実行するために最適化するオープンソースの機械学習コンパイラです。

  • プロジェクト
    mratsim/Arraymancer

    Nim 用の高性能 n 次元テンソルライブラリで、CPU、CUDA、OpenCL バックエンド上での科学計算、機械学習、深層学習のためのツールを提供します。

  • プロジェクト
    hasktorch/hasktorch

    PyTorch が共有するコア C++ ライブラリを活用する Haskell 用テンソルおよびニューラルネットワークライブラリ