For a library for speeding up LLM token generation, the first results are intel-analytics/bigdl (BigDL is a PyTorch acceleration framework that explicitly includes speculative decoding for LLM inference, fitting your search; but its broad focus on Intel hardware optimizations means details on draft model support and configurable trade-offs are not prominently described), pytorch-labs/gpt-fast and openvinotoolkit/openvino (OpenVINO is an inference engine that explicitly supports speculative decoding to accelerate LLM text generation, making it a direct tool for this purpose, though it is a broader platform rather than a dedicated speculative-decoding library). meta-pytorch/gpt-fast and huggingface/transformers round out the shortlist. Compare the match explanations and check the project documentation against your requirements.
High-performance libraries and implementations designed to accelerate large language model inference through speculative execution techniques.
BigDL is a PyTorch acceleration framework and distributed inference engine designed for large language models. It provides a toolkit for running models on Intel hardware, integrating quantization tools and libraries for parameter-efficient fine-tuning. The project distinguishes itself through the use of pipeline parallelism to distribute model workloads across multiple hardware accelerators. It utilizes low-bit integer quantization and speculative decoding to reduce memory footprints and decrease text generation latency. The system covers broad capabilities in model optimization, including w
BigDL is a PyTorch acceleration framework that explicitly includes speculative decoding for LLM inference, fitting your search; but its broad focus on Intel hardware optimizations means details on draft model support and configurable trade-offs are not prominently described.
gpt-fast is a PyTorch transformer inference engine designed for low-latency text generation. It functions as a distributed GPU inference library, a quantized model runner, and a speculative decoding framework. The system utilizes a speculative decoding workflow where a small draft model predicts token sequences for verification by a larger model to accelerate generation. It supports quantized model execution to reduce memory footprint and implements tensor parallelism to split computations across multiple GPUs. The project includes a standardized evaluation harness to measure the accuracy an
GPT-Fast is a dedicated speculative decoding framework for LLM inference acceleration, featuring draft-model verification, quantized execution, multi-GPU support, and a built-in evaluation harness—directly matching the core requirement while covering several requested capabilities.
OpenVINO is an AI inference engine and model serving platform designed to execute optimized deep learning models across CPUs, GPUs, and NPUs through a unified API. It includes a model optimization toolkit for converting, quantizing, and compressing models from various frameworks, alongside a specialized generative AI runtime for large language models. The project distinguishes itself through a plugin-based hardware acceleration layer that maps neural network operations to vendor-specific drivers. It features advanced execution mechanisms such as continuous batching, speculative decoding, and
OpenVINO is an inference engine that explicitly supports speculative decoding to accelerate LLM text generation, making it a direct tool for this purpose, though it is a broader platform rather than a dedicated speculative-decoding library.
gpt-fast is a PyTorch transformer inference engine designed for text generation using a native tensor library implementation. It provides a runtime for executing large language models without the need for external C++ extensions. The project implements speculative decoding to accelerate generation by using a small draft model for token prediction and a larger model for verification. It further optimizes performance through a compiled prefill stage and a multi-GPU tensor parallelism library that shards linear layers across multiple graphics processing units. Memory efficiency is managed throu
gpt-fast is a PyTorch inference engine that explicitly implements speculative decoding with a small draft model and verification, directly addressing the core need for accelerating LLM generation while being built for practical deployment.
Transformers is a comprehensive library for machine learning that provides a unified interface for training, fine-tuning, and deploying transformer-based models. It supports a wide range of tasks, including text classification, language modeling, question answering, and sequence-to-sequence translation, while offering specialized architectures for both text and vision processing. The framework includes tools for managing the entire model lifecycle, from data preprocessing and tokenization to distributed training and inference. The library features extensive support for model optimization and
Hugging Face Transformers is the leading open-source library for transformer models, and it directly supports speculative decoding through its "assisted generation" / prompt lookup decoding feature, allowing you to accelerate LLM text generation with a draft model and verification — exactly the kind of tool this search is after.
Lorax is a GPU-accelerated inference server and multi-adapter engine designed for serving large language models. It functions as a high-throughput system capable of deploying models via Kubernetes and managing the dynamic swapping of Low-Rank Adaptation adapters per request. The server distinguishes itself through multi-adapter dynamic batching, which allows requests using different adapter weights to be processed in a single GPU forward pass. It employs just-in-time adapter loading and weighted adapter merging to maximize throughput and enable multi-tasking without sacrificing performance.
Lorax is a GPU-accelerated LLM inference server that includes speculative decoding strategies, making it a relevant tool for accelerating text generation, though its primary focus is multi-adapter serving rather than a dedicated speculative decoding implementation.
llama-cpp-python provides a Python interface for the llama.cpp library, enabling the execution of large language models with hardware acceleration. It functions as a GGUF model loader and a structured text generator capable of running inference servers and multimodal runtimes for processing both text and image inputs. The project distinguishes itself through a local inference server that exposes model capabilities via an OpenAI-compatible web API. It supports advanced execution techniques including speculative decoding, weight quantization, and layer-based GPU offloading to manage memory acro
This repository wraps llama.cpp in Python and explicitly supports speculative decoding with draft models and verification, so it is a genuine tool in this category; however, it does not integrate directly with Hugging Face Transformers or include pre-built benchmarks, so it covers the core capability but not all requested features.
llama.cpp is a high-performance C++ inference engine and runtime for executing large language models locally across various hardware architectures. It provides the core components for local model execution, including a dedicated model quantizer for compressing weights into the GGUF format and a system for generating text embeddings for semantic search. The project distinguishes itself through specialized memory and execution optimizations, such as block-wise weight quantization to reduce memory footprints and memory-mapped model loading. It supports structured text generation by using formal
llama.cpp is a high-performance local LLM inference engine that includes speculative decoding strategies to accelerate generation, with configurable draft parameters and performance benchmarks, though its integration with Hugging Face Transformers is indirect via model conversion, making it a relevant tool for applying speculative decoding.
MiniCPM is a collection of small language models designed for local, on-device deployment in resource-constrained environments. The project focuses on running dense Transformer models on consumer hardware, including GPUs, CPUs, and Apple Silicon, without requiring custom code forks. The project distinguishes itself through heavy optimization for edge hardware, utilizing quantized weight compression in GGUF and MLX formats to reduce memory overhead. It implements advanced inference techniques such as speculative sampling and radix-tree prefix caching to accelerate generation speed and throughp
MiniCPM implements speculative sampling (a form of speculative decoding) and prefix caching to accelerate generation from its small language models, making it a relevant tool for applying speculative decoding to LLM inference, though it is tied to the MiniCPM model family rather than being a general-purpose library.
Intel XPU LLM Acceleration Library is a toolkit designed to accelerate large language model inference and finetuning on Intel CPUs, GPUs, and NPUs. It provides a distributed inference engine for scaling models across multiple accelerators, a multimodal model runtime for vision and speech tasks, and a low-bit model quantization tool for converting weights into INT4, FP8, and GGUF formats. The project features a parameter-efficient finetuning framework that enables model adaptation using QLoRA and DPO on Intel hardware. It distinguishes itself by providing specialized optimizations for Intel XP
Intel XPU LLM Acceleration Library is a general-purpose toolkit for accelerating LLM inference and finetuning on Intel hardware via quantization and distributed execution, but it does not implement the speculative decoding algorithm with draft models, rejection sampling, or the configurable trade-offs that this search targets.
FlashInfer is a library of high-performance GPU kernels purpose-built for accelerating large language model inference. It provides optimized implementations for attention operations (including flash attention, page attention, multi-head latent attention, and cascade attention) using paged key-value caches, fused kernel composition, and just-in-time compilation. The library also includes specialized kernels for mixture-of-experts layers, block-scaled low-precision quantization (FP8, FP4), and distributed collective communication. What distinguishes FlashInfer is its fused all-reduce communicat
FlashInfer is a library of GPU kernels for general LLM inference acceleration, but it does not implement speculative decoding, draft model support, or rejection sampling, so it is a building block for inference rather than a tool that applies speculative decoding.
Torchtune is a PyTorch-native library for fine-tuning, aligning, and quantizing large language models. It provides a configurable training pipeline orchestrated through YAML recipes, with CLI overrides and component swapping, distributed training via FSDP2, memory optimizations, and parameter-efficient fine-tuning methods like LoRA, DoRA, and QLoRA. The library distinguishes itself through its YAML-driven configuration system that defines all training parameters and instantiates components from config files, with full CLI override capability for any field or component at launch time. It suppo
Torchtune is a fine-tuning and alignment library for LLMs, not an inference acceleration tool, so it does not implement speculative decoding as you are looking for.
| Repository | Stars | Language | License | Last push |
|---|---|---|---|---|
| intel-analytics/bigdl | 8.8K | Python | Apache-2.0 | |
| pytorch-labs/gpt-fast | 6.2K | Python | BSD-3-Clause | |
| openvinotoolkit/openvino | 10.4K | C++ | Apache-2.0 | |
| meta-pytorch/gpt-fast | 6.2K | Python | BSD-3-Clause | |
| huggingface/transformers | 161.6K | Python | Apache-2.0 | |
| predibase/lorax | 3.7K | Python | apache-2.0 | |
| abetlen/llama-cpp-python | 10K | Python | mit | |
| ggerganov/llama.cpp | 116.9K | C++ | MIT | |
| openbmb/minicpm | 9.5K | Jupyter Notebook | Apache-2.0 | |
| intel/ipex-llm | 8.8K | Python | Apache-2.0 |