Repolode
keras-team

keras-team/keras-hub

Biblioteca de modelos pré-treinados para Keras 3, com modelos de texto, imagem e áudio que rodam em JAX, TensorFlow ou PyTorch.

Estrelas
996
Forks
377
Issues abertas
273
Último commit
há 3 dias

O que é?

KerasHub é uma biblioteca de modelagem com modelos pré-treinados, desenvolvida como extensão da API principal do Keras. Ela oferece implementações em Keras 3 de arquiteturas populares, além de checkpoints pré-treinados hospedados no Kaggle Models. Seus componentes Layer e Model abrangem tarefas como geração, classificação e outros fluxos de trabalho integrados para dados de texto, imagem e áudio.

Uma única definição de modelo funciona com JAX, TensorFlow e PyTorch, selecionados pela variável de ambiente KERAS_BACKEND. Os modelos podem ser ajustados em GPUs e TPUs, usar técnicas PEFT integradas em aceleradores individuais ou ser treinados em larga escala com paralelismo de modelo e de dados.

Quem já conhece o Keras pode começar rapidamente: carregar um preset com from_preset e chamar fit ou predict segue o fluxo de trabalho habitual do Keras. O projeto está em desenvolvimento ativo na fase pré-lançamento 0.y.z, então as APIs podem mudar.

Recursos principais

  • 01Implementações em Keras 3 de arquiteturas populares de modelos de texto, imagem e áudio.
  • 02Checkpoints pré-treinados que podem ser carregados pelo nome com from_preset.
  • 03Uma única definição de modelo roda em JAX, TensorFlow ou PyTorch.
  • 04Técnicas PEFT integradas para treinar em um único acelerador.
  • 05Oferece suporte a treinamento com paralelismo de modelo e de dados para ajustes em maior escala.

Casos de uso

  • Ajustar um classificador no estilo BERT pré-treinado em um conjunto de dados de texto personalizado.
  • Executar classificação de imagens com um checkpoint ResNet pré-treinado.
  • Adaptar grandes modelos de linguagem com fine-tuning eficiente em parâmetros quando a memória da GPU é limitada.
  • Migrar um pipeline de treinamento do Keras entre os backends JAX, TensorFlow e PyTorch.

Início rápido

  1. 1

    Instale o KerasHub pelo PyPI

    pip install --upgrade keras-hub
  2. 2

    Escolha um backend definindo a variável de ambiente antes de importar

    export KERAS_BACKEND=jax
  3. 3

    Importe o KerasHub e carregue um modelo pré-treinado de um preset

Antes de usar

  • A instalação do KerasHub sempre inclui o TensorFlow para o pré-processamento com tf.data.
  • A API é anterior à versão 1.0 e pode deixar de ser compatível a qualquer momento.
  • Alguns modelos pré-treinados incluídos são fornecidos por terceiros sob licenças separadas.

Em coleções

Tópicos

Perguntas frequentes

O que é keras-hub?
Biblioteca de modelos pré-treinados para Keras 3, com modelos de texto, imagem e áudio que rodam em JAX, TensorFlow ou PyTorch. KerasHub é uma biblioteca de modelagem com modelos pré-treinados, desenvolvida como extensão da API principal do Keras. Ela oferece implementações em Keras 3 de arquiteturas populares, além de checkpoints pré-treinados hospedados no Kaggle Models. Seus componentes Layer e Model abrangem tarefas como geração, classificação e outros fluxos de trabalho integrados para dados de texto, imagem e áudio.
keras-hub é gratuito? Qual é a licença?
keras-hub é open source sob a licença Apache-2.0. Confira os termos da licença antes de usá-lo em produtos comerciais.
keras-hub ainda é mantido?
Manutenção ativa. O último commit de keras-hub foi em 4 de nov. de 2026.
Para que posso usar keras-hub?
Ajustar um classificador no estilo BERT pré-treinado em um conjunto de dados de texto personalizado. Executar classificação de imagens com um checkpoint ResNet pré-treinado. Adaptar grandes modelos de linguagem com fine-tuning eficiente em parâmetros quando a memória da GPU é limitada. Migrar um pipeline de treinamento do Keras entre os backends JAX, TensorFlow e PyTorch.
Como instalar keras-hub?
Instale o KerasHub pelo PyPI: pip install --upgrade keras-hub Escolha um backend definindo a variável de ambiente antes de importar: export KERAS_BACKEND=jax Importe o KerasHub e carregue um modelo pré-treinado de um preset
Onde consigo keras-hub?
O código-fonte e as instruções de instalação estão no GitHub: https://github.com/keras-team/keras-hub

Resumo gerado por IA a partir do README do projeto. As estatísticas do GitHub são atualizadas diariamente.

Projetos relacionados