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) 技术，并支持大规模的模型和数据并行训练。