PJRT の例
bookmark_borderbookmark
コレクションでコンテンツを整理
必要に応じて、コンテンツの保存と分類を行います。
例: JAX CUDA プラグイン
- ラッパーを介した PJRT C API の実装(pjrt_c_api_gpu.h)。
- パッケージのエントリ ポイント(setup.py)を設定します。
- initialize() メソッドを実装します(__init__.py)。
- CUDA の任意の jax テストでテストできます。
フレームワークの実装
フレームワーク側で PJRT を使用して PJRT デバイスとインターフェースを構築するための参考資料:
- JAX
- jax-ml/jax は、
xla_client
API を介して PJRT API とやり取りします。
- GoMLX
- ZML
ハードウェアの実装
- 完全統合プラグイン(PJRT+MLIR+XLA):
- ライト インテグレーション プラグイン(PJRT+MLIR):
- StableHLO リファレンス インタープリタ プラグイン(MLIR ベースの C++ プラグイン。devlabs 後にリンク)
- Tenstorrent-XLA プラグイン(MLIR ベースの C プラグイン)
特に記載のない限り、このページのコンテンツはクリエイティブ・コモンズの表示 4.0 ライセンスにより使用許諾されます。コードサンプルは Apache 2.0 ライセンスにより使用許諾されます。詳しくは、Google Developers サイトのポリシーをご覧ください。Java は Oracle および関連会社の登録商標です。
最終更新日 2024-11-15 UTC。
[[["わかりやすい","easyToUnderstand","thumb-up"],["問題の解決に役立った","solvedMyProblem","thumb-up"],["その他","otherUp","thumb-up"]],[["必要な情報がない","missingTheInformationINeed","thumb-down"],["複雑すぎる / 手順が多すぎる","tooComplicatedTooManySteps","thumb-down"],["最新ではない","outOfDate","thumb-down"],["翻訳に関する問題","translationIssue","thumb-down"],["サンプル / コードに問題がある","samplesCodeIssue","thumb-down"],["その他","otherDown","thumb-down"]],["最終更新日 2024-11-15 UTC。"],[],[]]