NVIDIA、Python科学コミュニティ向け数学ライブラリ「nvmath-python」1.0 を一般公開

記事の3ポイント

1. 1

   CUDA-X ライブラリの Python 化と抽象化

   nvmath-python は cuFFT や cuBLASLt などの CUDA-X 数学ライブラリに対する Pythonic な抽象化レイヤーとして機能し、既存のワークフローを崩さずに GPU 性能を利用可能にする。

2. 2

   ユニバーサルスパーステンソル（UST）の実装

   独自のドメイン固有言語を通じてユーザーがアプリケーション最適化されたスパース形式を定義できる「ユニバーサルスパーステンソル」を採用し、従来の実装手法にない柔軟なアプローチを提供する。

3. 3

   多様な環境への迅速かつ柔軟なインストール

   pip や conda などのパッケージマネージャーを選べ、依存関係の自動解決や最小限のインストール、CPU バックエンドや NumPy/CuPy/PyTorch との連携を自由に選択できる。