keras-team/keras-hub

Pretrained model hub for Keras 3.

What it solves

KerasHubは、異なる機械学習フレームワーク間で事前学習済みモデルにアクセスし、使用するための、シンプルで柔軟な方法を提供します。JAX、TensorFlow、PyTorch の間で切り替える際に、モデルアーキテクチャや学習ループを書き直す必要がなくなります。

How it works

KerasHubは、コア Keras 3 API を拡張し、モデルアーキテクチャを Layer および Model の実装として提供します。これらを Kaggle Models で利用可能な事前学習済みチェックポイントと組み合わせます。これにより、ユーザーはプリセットからモデルをロードし、tf.data を使用して前処理を行い、任意のサポートされているバックエンドで推論またはファインチューニングを行うことができます。

Who it’s for

Keras を使用し、テキスト、画像、および音声データの事前学習済みモデルを、複数のフレームワークにわたるアプリケーションや研究プロジェクトに統合したい開発者や研究者。

Highlights

  • Multi-framework support: 単一の定義から JAX、TensorFlow、PyTorch で動作します。
  • Multimodal capabilities: 生成や分類などのタスクのために、テキスト、画像、および音声データのモデルをサポートします。
  • Cross-platform training: GPU や TPU でそのままファインチューニングが可能です。
  • Efficient fine-tuning: 内蔵のパラメータ効率の高いファインチューニング (PEFT) 技術と、大規模なモデルおよびデータ並列学習のサポートが含まれています。