A pretrained modeling library for Keras 3 offering simple, flexible, and fast access to models for text, image, and audio tasks.
KerasHub is a pretrained modeling library that provides Keras 3 implementations of popular model architectures, paired with pretrained checkpoints available on Kaggle Models. It solves the problem of accessing and fine-tuning state-of-the-art models for text, image, and audio tasks with a simple and flexible API. Developers can use it for tasks like classification, generation, and more, leveraging multi-framework support and scalable training options.
Machine learning engineers and researchers familiar with Keras who need quick access to pretrained models for prototyping, fine-tuning, or deploying models across JAX, TensorFlow, or PyTorch backends.
Developers choose KerasHub for its seamless integration with the Keras API, multi-framework compatibility from a single model definition, and access to a curated hub of pretrained models, reducing the complexity of implementing advanced deep learning models.
Pretrained model hub for Keras 3.
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Models support JAX, TensorFlow, and PyTorch from a single definition, enabling flexible training and inference across different hardware without code changes.
Components are provided as Layer and Model implementations, making them instantly familiar to Keras users and reducing the learning curve.
Offers access to pretrained checkpoints on Kaggle Models for tasks like image and text classification, allowing quick experimentation with state-of-the-art models.
Includes built-in PEFT techniques and support for model/data parallel training, facilitating efficient fine-tuning on large datasets and accelerators like GPUs and TPUs.
Installing KerasHub always pulls in TensorFlow for the tf.data API, even when using JAX or PyTorch backends, which can bloat environments and complicate deployment.
As a pre-release library (version 0.y.z), APIs are not stable and compatibility may break, making it risky for production use without careful version management.
The curated hub on Kaggle Models may not include all cutting-edge models, forcing users to wait for updates or rely on external libraries for unsupported architectures.