Shardy

Shardy は、すべての方言に対応した MLIR ベースのテンサー パーティショニング システムです。GSPMD チームと PartIR チームの両方のコラボレーションから構築されたこのシステムは、両方のシステムの長所と、両方のチームとユーザーの共有された経験を組み込んでいます。

スタートガイド

Shardy を使用するモデル デベロッパーは、JAX で Shardy を使用するをご覧ください。

Shardy を構成するコンポーネントの詳細については、Shardy の概要をご覧ください。

通信チャネル

  • 問題や機能リクエストは、openxla/shardy リポジトリに提出できます。
  • Shardy に関する質問は、OpenXLA Discord で受け付けています。

ステータス

Shardy は現在も開発中です。コア ダイアレクトと C バインディングは完全に機能します。Python バインディングは現在開発中です。