Framework de deep learning baseado em PyTorch que automatiza loops de treinamento, escalabilidade para múltiplas GPUs e código de infraestrutura, mantendo o controle total do código do modelo.
basetenlabs/truss
CLI e toolkit Python para empacotar e implantar modelos de ML como servidores de modelos em produção na Baseten ou na sua própria infraestrutura.
- Estrelas
- 1,231
- Forks
- 142
- Issues abertas
- 90
- Último commit
- há 23 horas
O que é?
O Truss empacota a lógica de serving, os pesos e as dependências de um modelo em um servidor de modelos pronto para ser executado em um container e, em seguida, implanta esse servidor na Baseten ou na sua própria infraestrutura. O fluxo é controlado por uma classe de modelo em Python para código personalizado e por um arquivo config.yaml que declara o hardware, como o tipo de GPU, e as opções de build. Os desenvolvedores podem iterar com um modo de desenvolvimento com recarga automática, enquanto a configuração do Docker e do Kubernetes é feita pelo Truss.
O projeto oferece suporte a modelos de frameworks open source populares, incluindo vLLM, SGLang, TensorRT-LLM, transformers, diffusers, PyTorch e TensorFlow. Para LLMs, um arquivo de configuração pode criar um engine TensorRT-LLM com opções como quantização FP8 e comprimento máximo de sequência, e a implantação resultante disponibiliza uma API compatível com OpenAI. Os recursos para produção na Baseten incluem configuração de GPU, secrets, cache e autoscaling.
É indicado para equipes que querem transformar um modelo do Hugging Face ou um modelo PyTorch personalizado em um endpoint hospedado sem escrever a infraestrutura de serving. O repositório também inclui um schema JSON para config.yaml que habilita o preenchimento automático no editor.
Recursos principais
- 01Empacota o código, os pesos e as dependências do modelo em um servidor de modelos
- 02Modo de desenvolvimento com recarga automática para iterar mais rápido
- 03Cria engines TensorRT-LLM com quantização FP8 a partir de um arquivo de configuração YAML
- 04Disponibiliza endpoints compatíveis com OpenAI para implantações de LLMs
- 05Oferece suporte a modelos vLLM, SGLang, transformers, diffusers, PyTorch e TensorFlow
- 06Oferece configuração de GPU, secrets, cache e autoscaling na Baseten
Casos de uso
- Implantar um LLM do Hugging Face como endpoint de API hospedado sem escrever Dockerfiles
- Servir em produção um modelo PyTorch personalizado ou de difusão com aceleração por GPU
- Substituir stacks de serving de modelos feitas manualmente por um fluxo orientado por configuração
Início rápido
- 1
Instale o Truss com pip
pip install --upgrade truss - 2
Faça login no Baseten
uvx truss login - 3
Crie a estrutura de um novo projeto
uvx truss init qwen-2.5-3b && cd qwen-2.5-3b - 4
Implante o modelo no Baseten
uvx truss push
Antes de usar
- Para implantar o início rápido, é necessário ter uma conta e uma chave de API do Baseten.
- A implantação de TRT-LLM do início rápido requer uma GPU L4 no Baseten, e o modo --watch não é compatível com TRT-LLM.
Em coleções
Tópicos
Perguntas frequentes
- O que é truss?
- CLI e toolkit Python para empacotar e implantar modelos de ML como servidores de modelos em produção na Baseten ou na sua própria infraestrutura. O Truss empacota a lógica de serving, os pesos e as dependências de um modelo em um servidor de modelos pronto para ser executado em um container e, em seguida, implanta esse servidor na Baseten ou na sua própria infraestrutura. O fluxo é controlado por uma classe de modelo em Python para código personalizado e por um arquivo config.yaml que declara o hardware, como o tipo de GPU, e as opções de build. Os desenvolvedores podem iterar com um modo de desenvolvimento com recarga automática, enquanto a configuração do Docker e do Kubernetes é feita pelo Truss.
- truss é gratuito? Qual é a licença?
- truss é open source sob a licença MIT. Confira os termos da licença antes de usá-lo em produtos comerciais.
- truss ainda é mantido?
- Manutenção ativa. O último commit de truss foi em 6 de nov. de 2026.
- Para que posso usar truss?
- Implantar um LLM do Hugging Face como endpoint de API hospedado sem escrever Dockerfiles Servir em produção um modelo PyTorch personalizado ou de difusão com aceleração por GPU Substituir stacks de serving de modelos feitas manualmente por um fluxo orientado por configuração
- Como instalar truss?
- Instale o Truss com pip: pip install --upgrade truss Faça login no Baseten: uvx truss login Crie a estrutura de um novo projeto: uvx truss init qwen-2.5-3b && cd qwen-2.5-3b Implante o modelo no Baseten: uvx truss push
- Onde consigo truss?
- O código-fonte e as instruções de instalação estão no GitHub: https://github.com/basetenlabs/truss
Resumo gerado por IA a partir do README do projeto. As estatísticas do GitHub são atualizadas diariamente.
Projetos relacionados
Servidor de deep learning, pacote Python wheel e CLI para treinar e executar modelos com imagens, textos, dados tabulares e séries temporais por meio de uma API REST.
Modular Platform: um espaço open source para o framework de IA MAX e a linguagem de programação Mojo, voltado ao desenvolvimento e à implantação de modelos de IA.
Framework declarativo de deep learning para treinar, ajustar e implantar LLMs e modelos multimodais e tabulares a partir de configurações YAML.
Solução independente de serving de modelos, baseada em Java, que disponibiliza modelos de PyTorch, TensorFlow, ONNX e outros via endpoints HTTP.
Servidor de inferência em C++ que disponibiliza modelos de IA generativa e deep learning clássico otimizados para OpenVINO por meio de APIs compatíveis com OpenAI e KServe.