Google/JAX にも Keras APIが

Vengineerの戯言 : Twitter
SystemVerilogの世界へようこそ、すべては、SystemC v0.9公開から始まった

Keras開発者のFrançois Cholletさんの下記のツイートから Google/JAX にも Keras API 付きそうですね。
記録のために、引用します。

It seems not many people know about it today, but the people I know who've tried it love it.

Maybe we can bring the Keras API to JAX as an extension in the future

XLAへのブリッジは、ここ、xla_bridge.py なんですよ。