OpenTPU:GoogleのTPUのオープンソース再実装
2025-05-28
カリフォルニア大学サンタバーバラ校のArchLabは、GoogleのTensor Processing Unit(TPU)のオープンソース再実装であるOpenTPUをリリースしました。Googleの論文に基づいていますが、正式な仕様がないため、Googleの設計とはいくつかの実装の詳細で異なります。現在、行列乗算とReLU/シグモイド活性化関数をサポートしていますが、畳み込みやプーリングなどの機能は欠けています。PyRTLを使用して実装されており、ハードウェアシミュレータと機能シミュレータ、結果検証のためのチェッカーが含まれています。このアルファリリースでは、完全な256x256 OpenTPUの具体的な合成数値はありませんが、オープンソースであるため、研究者にとって貴重な学習と改善の機会を提供します。
ハードウェア