Vengineerの妄想

人生を妄想しています。

Tenstorent の tt-xla

はじめに

Tenstorrentの tt-xla

Googleの下記の「PJRT: Simplifying ML Hardware and Framework Integration」に対応しようとしているようです。

opensource.googleblog.com

これができると、

  • Tensorflow
  • JAX
  • PyTorch

が動くようになるようですね。

OpenXLA

OpenXLA については、ここにありますね。

OpenXLAの関連ブログ

vengineer.hatenablog.com

vengineer.hatenablog.com

tt-xla

github.com

内部的には、

を使うようです。

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」

www.youtube.com

「Building a PJRT Plugin for a Custom Accelerator」

www.youtube.com

なるビデオがアップされていました。