Vengineerの戯言

人生は短いけど、長いです。人生を楽しみましょう!

「TensorFlow XLAの衝撃」から6年、OpenXLAとして TensorFlow, PyTorch, JAX の基盤となるの?

はじめに

GoogleからOpenXLAのアナウンスがありました。

opensource.googleblog.com

下図は上記のブログにある図です。説明のために引用します。TensorFlow, PyTorch, JAX から StableHLO にし、Target-Independ-Optimizer にて最適化し、Hardware Optimizations & Targets の流れになるようです。

この図で対応するTargetは、

なんで、Arm CPUs だけ、s が付いているんでしょうか。。。。

Youtube & スライド

ここに、OpenXLA に関するビデオがあります。

www.youtube.com

スライドも見つけました。

www.linkedin.com

Tensorflowとの関係

tensorflow/tensorflow/compiler/xlaxla/xla になった感じですね。

おわりに

XLAを知ったのが、2017年の春、2017年2月20日の「XLAの衝撃」

TensorFlow XLAの衝撃 - Vengineerの戯言

これであたしの人生が変わりました。。。

それから6年、OpenXLA になったという感じですね。

なんか、未来が見えていたのかもしれません。。。*1

*1: