はじめに
Tenstorrentの tt-xla
Googleの下記の「PJRT: Simplifying ML Hardware and Framework Integration」に対応しようとしているようです。
これができると、
- Tensorflow
- JAX
- PyTorch
が動くようになるようですね。
OpenXLA
OpenXLA については、ここにありますね。
OpenXLAの関連ブログ
tt-xla
内部的には、
- tt-mlir
- stableHLO
を使うようです。
This project is currently only supported for nebula boards and does not provide support for galaxy boards.
とあるように、
- nebula
で利用可能のようです。
おわりに
- TensorFLow
- JAX
- PyTorch
が動けば、ユーザーも増えますかね。。。。。
Youtubeに、
「OpenXLA Dev Lab 2024 Nov 14 2024」
と
「Building a PJRT Plugin for a Custom Accelerator」
なるビデオがアップされていました。