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: 텍s, 이미지, 오디오 데이터에 대한 생성 및 분류 작업을 위한 모델을 지원합니다.
- Cross-platform training: GPU와 TPU에서 즉시 미세 조정을 수행할 수 있습니다.
- Efficient fine-tuning: 내장된 Parameter-Efficient Fine-Tuning (PEFT) 기술과 대규모 모델 및 데이터 병렬 학습 지원을 포함합니다.