Luce-Org/lucebox
LLM speculative inference server for heterogeneous hardware & consumer GPUs
What it solves
Lucebox is a high-performance inference engine designed to bring fast, local LLM execution to heterogeneous consumer hardware, including NVIDIA GPUs, AMD GPUs, and APUs (like Strix Halo). It addresses the performance bottlenecks of running large models on consumer-grade equipment by using speculative inference and custom hardware-tuned kernels.
How it works
The engine employs several advanced optimization techniques to maximize throughput and reduce latency:
- Speculative Inference: Uses smaller "drafter" models to predict tokens, which are then verified by the larger target model, significantly speeding up decoding.
- Custom Kernels: Implements specialized kernels (e.g., DFlash, PFlash, KVFlash, Megakernel) tuned for specific hardware architectures (CUDA and HIP).
- Memory Management: Utilizes paged attention and continuous batching to handle multiple concurrent requests efficiently.
- Heterogeneous Execution: Supports distributing model execution across different types of hardware (e.g., combining an AMD GPU and an APU).
Who it’s for
It is intended for developers and AI enthusiasts who want to run powerful LLMs locally on consumer hardware with high token-per-second rates, as well as those integrating LLMs into coding clients via its OpenAI-compatible API.
Highlights
- Broad Hardware Support: Compatible with NVIDIA (Ampere, Ada, Blackwell) and AMD (RDNA3, RDNA4) architectures.
- High Throughput: Achieves significant speedups (e.g., over 200 tok/s on certain AMD hardware).
- OpenAI-Compatible API: Allows easy integration with various coding clients and harnesses.
- Speculative Prefill and Decoding: Optimizes both the initial prompt processing and the token generation phase.
- ** uma-GPU and Mixed-GPU Profiles**: Provides pre-configured setups for various hardware combinations.
Related
- Project
- Project
- Project
- Project
- Project