Keras Core 0.1 : 新しいマルチバックエンド Keras (README) (翻訳/解説)

【深層学習：Keras Core 0.1】新しいマルチバックエンド Keras

Keras API の新しいマルチバックエンド実装である Keras Core が公開されました。

既に幾つかドキュメントが用意されていますので、順次翻訳していきます。最初は README です。

Keras 3.0 のプレビュー版という位置づけで tf.keras の置き換えとなりますが、TensorFlow, JAX, 更には PyTorch (!) をサポートします。

これは上位の Keras ワークフローを任意のフレームワークの上で実行できるだけでなく、各フレームワークの優位点を自在に利用できることを意味します。

例えば JAX のスケーラビリティやパフォーマンスを利用したり、tf.data.Dataset パイプラインの代わりに PyTorch DataLoader を使用して訓練することもできます。

Keras モデルを native TF, JAX や PyTorch でゼロから書かれた訓練ループで訓練することができます。

Keras モデルを PyTorch-native Module や JAX-native モデル関数の一部として使用することもできます。
[More]