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 上可用的預訓練檢查點（checkpoints）配對。這使得使用者可以從預設值中載入模型，使用 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) 技術，並支援大規模的模型和數據並行訓練。