Implementação em PyTorch do GeoCLIP, um modelo no estilo CLIP que alinha imagens a coordenadas GPS para geolocalização global de imagens.
ludwig-ai/ludwig
Framework declarativo de deep learning para treinar, ajustar e implantar LLMs e modelos multimodais e tabulares a partir de configurações YAML.
- Estrelas
- 11,793
- Forks
- 1,213
- Issues abertas
- 3
- Último commit
- há 5 dias
O que é?
O Ludwig permite criar modelos de IA escrevendo uma configuração declarativa em YAML, em vez de implementar um loop de treinamento, um pipeline de dados e código de avaliação personalizados. Você lista os recursos de entrada e saída com seus tipos (texto, imagem, número, áudio, séries temporais e outros), e o framework monta encoders, combinadores e decoders; em seguida, executa treinamento, avaliação e inferência pela linha de comando.
Para modelos de linguagem, o Ludwig oferece fine-tuning supervisionado, alinhamento com DPO, KTO, ORPO e GRPO, adaptadores LoRA, DoRA e outros PEFT, QLoRA de 4 bits e quantização com torchao, além de fine-tuning de modelos visão-linguagem. Modelos tabulares e multimodais podem usar qualquer encoder de texto do HuggingFace ou encoder de imagem do timm, com busca de hiperparâmetros via Ray Tune ou Optuna.
Os modelos treinados podem ser disponibilizados por um endpoint REST com FastAPI, vLLM, Ray Serve ou KServe; exportados como bundles SafeTensors, ONNX ou torch.export; ou enviados ao HuggingFace Hub. O projeto é construído sobre PyTorch, Transformers e Ray e pode ser executado de um laptop a clusters com várias GPUs ou Ray, bastando alterar a configuração do backend.
Recursos principais
- 01Defina modelos e execuções de treinamento inteiramente em YAML, sem implementar um loop de treinamento personalizado.
- 02Faça fine-tuning de LLMs com SFT, DPO, KTO, ORPO e GRPO, incluindo LoRA e QLoRA.
- 03Combine entradas de texto, imagem, dados numéricos, áudio e séries temporais em um único modelo.
- 04Faça busca de hiperparâmetros com Ray Tune ou Optuna.
- 05Disponibilize modelos via REST, vLLM, Ray Serve ou KServe e exporte-os para ONNX ou SafeTensors.
Casos de uso
- Fazer fine-tuning de um LLM de pesos abertos com dados de instrução usando LoRA e pouca memória de GPU.
- Criar um classificador ou regressor tabular sem escrever código de treinamento.
- Treinar um modelo que combine imagens de produtos, texto de avaliações e campos numéricos.
- Prever séries temporais com encoders PatchTST ou N-BEATS.
Início rápido
- 1
Instale o Ludwig pelo PyPI (Python 3.12 ou superior).
pip install ludwig - 2
Escreva uma configuração YAML que descreva os recursos de entrada e saída do seu modelo.
- 3
Treine o modelo com seu conjunto de dados.
ludwig train --config model.yaml --dataset my_data.csv - 4
Execute previsões com o modelo treinado.
ludwig predict --model_path results/experiment_run/model --dataset new_data.csv
Antes de usar
- Requer Python 3.12 ou superior.
- O fine-tuning de LLMs grandes exige muitos recursos de GPU e pode exigir um token do Hugging Face para modelos com acesso restrito.
Em coleções
Tópicos
Perguntas frequentes
- O que é ludwig?
- Framework declarativo de deep learning para treinar, ajustar e implantar LLMs e modelos multimodais e tabulares a partir de configurações YAML. O Ludwig permite criar modelos de IA escrevendo uma configuração declarativa em YAML, em vez de implementar um loop de treinamento, um pipeline de dados e código de avaliação personalizados. Você lista os recursos de entrada e saída com seus tipos (texto, imagem, número, áudio, séries temporais e outros), e o framework monta encoders, combinadores e decoders; em seguida, executa treinamento, avaliação e inferência pela linha de comando.
- ludwig é gratuito? Qual é a licença?
- ludwig é open source sob a licença Apache-2.0. Confira os termos da licença antes de usá-lo em produtos comerciais.
- ludwig ainda é mantido?
- Manutenção ativa. O último commit de ludwig foi em 2 de nov. de 2026.
- Para que posso usar ludwig?
- Fazer fine-tuning de um LLM de pesos abertos com dados de instrução usando LoRA e pouca memória de GPU. Criar um classificador ou regressor tabular sem escrever código de treinamento. Treinar um modelo que combine imagens de produtos, texto de avaliações e campos numéricos. Prever séries temporais com encoders PatchTST ou N-BEATS.
- Como instalar ludwig?
- Instale o Ludwig pelo PyPI (Python 3.12 ou superior).: pip install ludwig Escreva uma configuração YAML que descreva os recursos de entrada e saída do seu modelo. Treine o modelo com seu conjunto de dados.: ludwig train --config model.yaml --dataset my_data.csv Execute previsões com o modelo treinado.: ludwig predict --model_path results/experiment_run/model --dataset new_data.csv
- Onde consigo ludwig?
- O código-fonte e as instruções de instalação estão no GitHub: https://github.com/ludwig-ai/ludwig
Resumo gerado por IA a partir do README do projeto. As estatísticas do GitHub são atualizadas diariamente.
Projetos relacionados
Biblioteca de modelos pré-treinados para Keras 3, com modelos de texto, imagem e áudio que rodam em JAX, TensorFlow ou PyTorch.
Um LLM de vídeo treinado com transcrição de fala em streaming que gera comentários em tempo real sobre vídeos ao vivo, apresentado no artigo LiveCC da CVPR 2025.
Biblioteca de deep learning baseada em PyTorch, com componentes de alto nível para treinar modelos rapidamente e recursos de baixo nível para pesquisa personalizada.
Biblioteca diferenciável de visão computacional baseada em PyTorch, com processamento de imagens otimizado para GPU, geometria, augmentations e operadores de correspondência de features.
Biblioteca Python para executar, treinar e avaliar modelos de IA do hub ModelScope por meio de uma interface unificada de pipeline e Trainer.