Repolode
Enviar
google-research

google-research/kauldron

Biblioteca modular e escalável para treinar modelos de machine learning em JAX/Flax, criada para acelerar a experimentação em pesquisa.

Ver no GitHubSite
Estrelas
359
Forks
36
Issues abertas
83
Último commit
ontem

O que é?

Kauldron é uma biblioteca Python para treinar modelos de machine learning, projetada para acelerar a pesquisa e oferecer modularidade. Seus componentes são independentes e podem ser usados fora do framework, permitindo que as equipes os combinem com seus próprios pipelines de dados, modelos Flax e otimizadores.

A biblioteca conecta conjuntos de dados como TFDS, Grain e SeqIO a qualquer modelo Flax e qualquer otimizador. As configurações podem ser personalizadas e sobrescritas, por exemplo, para testar diferentes arquiteturas de modelo ou alterar parâmetros de camadas internas. Configurações de exemplo, como a de um autoencoder para MNIST, servem como ponto de partida.

A biblioteca foi criada para um fluxo de trabalho com foco no Colab, que facilita a prototipagem rápida. Entre os recursos integrados estão integração com ferramentas de plotagem, um profiler, depuração post-mortem e verificação de shapes em tempo de execução.

Recursos principais

  • 01Funciona com TFDS, Grain, SeqIO ou pipelines de dados personalizados.
  • 02Aceita qualquer modelo Flax e qualquer otimizador.
  • 03As configurações podem ser testadas em sweeps e sobrescritas em qualquer nível.
  • 04Os componentes podem ser usados de forma independente, fora do framework.
  • 05Inclui um profiler e verificação de shapes em tempo de execução.

Casos de uso

  • Prototipar novas arquiteturas de modelo em notebooks do Colab.
  • Executar sweeps de hiperparâmetros ou de arquiteturas em pesquisas de machine learning.
  • Treinar modelos Flax com conjuntos de dados personalizados usando uma configuração padronizada.

Antes de usar

  • O README informa que este não é um produto com suporte oficial do Google.
  • O README não inclui instruções de instalação nem de execução.

Tópicos

#deep-learning#jax#flax#machine-learning#training-framework#python

Perguntas frequentes

O que é kauldron?
Biblioteca modular e escalável para treinar modelos de machine learning em JAX/Flax, criada para acelerar a experimentação em pesquisa. Kauldron é uma biblioteca Python para treinar modelos de machine learning, projetada para acelerar a pesquisa e oferecer modularidade. Seus componentes são independentes e podem ser usados fora do framework, permitindo que as equipes os combinem com seus próprios pipelines de dados, modelos Flax e otimizadores.
kauldron é gratuito? Qual é a licença?
kauldron é open source sob a licença Apache-2.0. Confira os termos da licença antes de usá-lo em produtos comerciais.
kauldron ainda é mantido?
Manutenção ativa. O último commit de kauldron foi em 6 de nov. de 2026.
Para que posso usar kauldron?
Prototipar novas arquiteturas de modelo em notebooks do Colab. Executar sweeps de hiperparâmetros ou de arquiteturas em pesquisas de machine learning. Treinar modelos Flax com conjuntos de dados personalizados usando uma configuração padronizada.
Onde consigo kauldron?
O código-fonte e as instruções de instalação estão no GitHub: https://github.com/google-research/kauldron

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

Projetos relacionados