about
Petals: Collaborative Inference and Fine-Tuning of Large Models (arxiv.org)
2 points by jonbaer on Mar 17, 2023 | hide | past | pdf | discuss on HN

In plain words: Volunteers split a giant language model across many graphics cards, passing text through them so anyone can use it cheaply. It runs one of the largest models at about one step per second, beating spilling to ordinary memory, and lets users fine-tune and share add-ons.

Abstract · Petals: Collaborative Inference and Fine-tuning of Large Models

Many NLP tasks benefit from using large language models (LLMs) that often have more than 100 billion parameters. With the release of BLOOM-176B and OPT-175B, everyone can download pretrained models of this scale. Still, using these models requires high-end hardware unavailable to many researchers. In some cases, LLMs can be used more affordably via RAM offloading or hosted APIs. However, these techniques have innate limitations: offloading is too slow for interactive inference, while APIs are not flexible enough for research that requires access to weights, attention or logits. In this work, we propose Petals - a system for inference and fine-tuning of large models collaboratively by joining the resources of multiple parties. We demonstrate that this strategy outperforms offloading for very large models, running inference of BLOOM-176B on consumer GPUs with $\approx$ 1 step per second, which is enough for many interactive LLM applications. Unlike most inference APIs, Petals also natively exposes hidden states of served models, allowing to train and share custom model extensions based on efficient fine-tuning methods.

Alexander Borzunov, Dmitry Baranchuk, Tim Dettmers, Max Ryabinin, Younes Belkada, Artem Chumachenko, Pavel Samygin, Colin Raffel
arXiv:2209.01188 · cs.LG, cs.DC · submitted Sep 2, 2022 · updated Mar 2, 2023
abstract · pdf · html · 10 pages, 4 figures. The version 2 updates the benchmarks and the description of the chat application. Source code and docs: https://petals.ml

add comment on HN
Also discussed: Jul 2026 (2 points, 0 comments)