Repolode
keras-team

keras-team/keras-hub

Biblioteca de modelos preentrenados para Keras 3, con modelos de texto, imagen y audio compatibles con JAX, TensorFlow y PyTorch.

Estrellas
996
Forks
377
Issues abiertos
273
Último commit
hace 3 días

¿Qué es?

KerasHub es una biblioteca de modelado preentrenado creada como extensión de la API principal de Keras. Ofrece implementaciones para Keras 3 de arquitecturas de modelos populares, junto con checkpoints preentrenados alojados en Kaggle Models. Sus componentes Layer y Model cubren tareas como generación, clasificación y otros flujos de trabajo integrados para datos de texto, imagen y audio.

Una misma definición de modelo funciona con JAX, TensorFlow y PyTorch; el backend se selecciona mediante la variable de entorno KERAS_BACKEND. Los modelos se pueden ajustar en GPUs y TPUs, usar técnicas PEFT integradas en aceleradores individuales o entrenar a gran escala con paralelismo de modelos y de datos.

Los desarrolladores que ya conocen Keras pueden empezar rápidamente: cargar un preset con from_preset y llamar a fit o predict sigue el flujo de trabajo habitual de Keras. El proyecto está en desarrollo activo en la fase preliminar 0.y.z, por lo que las API pueden cambiar.

Funciones clave

  • 01Implementaciones para Keras 3 de arquitecturas populares de modelos de texto, imagen y audio.
  • 02Checkpoints preentrenados que se pueden cargar por nombre mediante from_preset.
  • 03Una misma definición de modelo funciona con JAX, TensorFlow o PyTorch.
  • 04Técnicas PEFT integradas para entrenar en un solo acelerador.
  • 05Admite entrenamiento con paralelismo de modelos y de datos para ajustar modelos a mayor escala.

Casos de uso

  • Ajustar un clasificador de tipo BERT preentrenado con un conjunto de datos de texto personalizado.
  • Ejecutar clasificación de imágenes con un checkpoint preentrenado de ResNet.
  • Adaptar modelos de lenguaje grandes mediante ajuste eficiente en parámetros cuando la memoria de la GPU es limitada.
  • Migrar un flujo de entrenamiento de Keras entre los backends JAX, TensorFlow y PyTorch.

Inicio rápido

  1. 1

    Instala KerasHub desde PyPI

    pip install --upgrade keras-hub
  2. 2

    Elige un backend configurando la variable de entorno antes de importar

    export KERAS_BACKEND=jax
  3. 3

    Importa KerasHub y carga un modelo preentrenado desde un preset

Antes de usarlo

  • Al instalar KerasHub, TensorFlow se instala siempre para el preprocesamiento con tf.data.
  • La API es anterior a la versión 1.0 y la compatibilidad puede romperse en cualquier momento.
  • Algunos modelos preentrenados incluidos los proporcionan terceros con licencias independientes.

En colecciones

Temas

Preguntas frecuentes

¿Qué es keras-hub?
Biblioteca de modelos preentrenados para Keras 3, con modelos de texto, imagen y audio compatibles con JAX, TensorFlow y PyTorch. KerasHub es una biblioteca de modelado preentrenado creada como extensión de la API principal de Keras. Ofrece implementaciones para Keras 3 de arquitecturas de modelos populares, junto con checkpoints preentrenados alojados en Kaggle Models. Sus componentes Layer y Model cubren tareas como generación, clasificación y otros flujos de trabajo integrados para datos de texto, imagen y audio.
¿keras-hub es gratuito? ¿Qué licencia tiene?
keras-hub es open source bajo la licencia Apache-2.0. Revisa los términos de la licencia antes de usarlo en productos comerciales.
¿keras-hub sigue mantenido?
Mantenimiento activo. El último commit de keras-hub fue el 4 nov 2026.
¿Para qué puedo usar keras-hub?
Ajustar un clasificador de tipo BERT preentrenado con un conjunto de datos de texto personalizado. Ejecutar clasificación de imágenes con un checkpoint preentrenado de ResNet. Adaptar modelos de lenguaje grandes mediante ajuste eficiente en parámetros cuando la memoria de la GPU es limitada. Migrar un flujo de entrenamiento de Keras entre los backends JAX, TensorFlow y PyTorch.
¿Cómo se instala keras-hub?
Instala KerasHub desde PyPI: pip install --upgrade keras-hub Elige un backend configurando la variable de entorno antes de importar: export KERAS_BACKEND=jax Importa KerasHub y carga un modelo preentrenado desde un preset
¿Dónde consigo keras-hub?
El código fuente y las instrucciones de instalación están en GitHub: https://github.com/keras-team/keras-hub

Resumen generado por IA a partir del README del proyecto. Las estadísticas de GitHub se actualizan a diario.

Proyectos relacionados