sten

Sten：Pytorchの効率的なスパースのためのインターフェイス

Stenは、Pytorch 1.11のスパースの現在の実装（Torch.sparse）で未回答のままである以下の質問を解決することを目的としています。

• モデルにスパース化戦略を組み込み、ランタイムに使用する方法は？
• トレーニング中にスパースレベルを同じに保つ方法は？
• フルオートグラードサポートを有効にする方法は？
• カスタムスパースフォーマットとオペレーターの実装を有効にする方法は？

例

• build_from_scratch.ipynbをチェックして、インターフェイスの使用の例を確認して、Pytorchモジュールをゼロから構築します。
• Modify_existing.ipynbを確認して、既存の密なPytorchモジュールをスパースに変換する例を確認してください。
• custom_implementations.ipynbを確認して、入力テンソルと出力テンソルの特定の形式に一致するスパルシファイと演算子のカスタム実装を登録する例を確認します。

クイックスタート

 git clone https://github.com/spcl/sten.git
cd sten
python -m venv venv
source venv/bin/activate
pip install .
python tests/test_api.py

インストール

 pip install sten

コード組織

コア実装はsten.pyにあります。 Jupyterノートブックの例は、例のディレクトリにあります。さらに多くの例は、テストディレクトリのテストの形で見つけることができます。テストは、プロジェクトルートでpytest呼び出すことで実行できます。