KerasFormers

概要

• KerasFormersは、Keras 3で実装された事前学習済みTransformerモデルのライブラリで、JAX・PyTorch・TensorFlowの各バックエンドで実行できます。
• Hugging Faceで公開されている現代的なモデルを、バックエンドごとに大幅な個別実装を行わず、Keras 3の共通APIで扱えるようにします。

主な特徴

• 画像、音声、LLM、VLMを含む60以上のアーキテクチャを収録し、幅広い生成AI・機械学習用途に対応しています。
• Hugging FaceのチェックポイントからKeras形式への重み変換と、各実装の数値的なパリティ検証を組み込んでいます。
• 既存のPyTorchコードを単にラップするのではなく、モデル構造やテンソル配置、Keras 3 Opsを考慮したバックエンド非依存の実装を目指しています。

こんな人におすすめ

• Keras 3を使って、TensorFlowだけでなくJAXやPyTorchでも同じモデルコードを動かしたい研究者・開発者に適しています。
• Hugging Faceの事前学習済みモデルをKeras環境へ移行し、重み変換や実装の再現性を効率よく検証したい人に向いています。