#170 · Primary category: Inference & Local Deploy

Medusa

llm llm-inference

Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Heads

Project last updated:06/25/24

GitHub Stars

2.8K

Forks

205

Contributors

11

License

Apache-2.0

Why we included this project

Most token-generation speedups lean on a second, smaller draft model that has to be trained, tuned, and kept in sync. Medusa sidesteps that entirely: you train a few extra lightweight decoding heads on the model you already have, leaving the original weights untouched and your distributed setup about the same. That means less operational overhead and no extra model to babysit, and the authors' technical report shows 2.2-3.6x speedups on several common LLMs. It works for both greedy and sampling-based generation, so it's a practical option for teams serving fine-tuned models who want lower latency without changing their serving stack.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category