pymc-devs/pytensor

何を解決するか

PyTensorは、多次元配列を含む複雑な数学的式を定義・最適化・評価する方法を提供します。PyMCの計算バックエンドとして機能し、効率的な数値計算と勾配計算を可能にします。

動作方法

シンボリックグラフフレームワークを使用して数学的演算を表現します。一部の現代的なディープラーニングライブラリとは異なり、PyTensorは最適化のためのインプレース変更が可能な静的グラフを維持します。このグラフをC、JAX、またはNumbaを介して実行可能コードにトランスパイルすることで、高いパフォーマンスを確保します。

対象ユーザー

確率的プログラミングやカスタムのシンボリック最適化に取り組む開発者や研究者向けに設計されており、柔軟でカスタマイズ可能なシンボリック数学計算のバックエンドが必要な人にとって最適です。

特徴

• 静的グラフ最適化: 不要な演算を削除するために、式グラフをインプレースで変更可能。
• マルチバックエンドコンパイル: C、JAX、Numbaを介したコンパイルをサポート。
• 自動微分: シンボリック式の勾配を計算可能。
• 拡張可能なフレームワーク: 純Pythonで構成されており、カスタム演算子の迅速な開発が可能。