vLLM CPU inference engine (AVX512 + VNNI + BF16 optimized)
Project description
Easy, fast, and cheap LLM serving for everyone
Buy Me a Coffee
Your support encourages me to keep creating/supporting my open-source projects. If you found value in this project, you can buy me a coffee to keep me up all the sleepless nights.
About
vLLM is a fast and easy-to-use library for LLM inference and serving. This PyPl package has only supports AVX512+VNNI+AVX512BF16. No support for AMXBF16 is available in this package. CPU inference will have the above available instruction set accelerations.
Originally developed in the Sky Computing Lab at UC Berkeley, vLLM has evolved into a community-driven project with contributions from both academia and industry.
vLLM is fast with:
- State-of-the-art serving throughput
- Efficient management of attention key and value memory with PagedAttention
- Continuous batching of incoming requests
- Fast model execution with AVX512+VNNI+AVX512BF16 on supported CPUs Use this package ONLY IF your CPU have avx512bf16 or newer instruction sets.
- Quantizations: GPTQ, AWQ, AutoRound, INT4, INT8, and FP8
- Optimized CPU kernels, including integration with FlashAttention and FlashInfer
- Speculative decoding
- Chunked prefill
vLLM is flexible and easy to use with:
- Seamless integration with popular Hugging Face models
- High-throughput serving with various decoding algorithms, including parallel sampling, beam search, and more
- Tensor, pipeline, data and expert parallelism support for distributed inference
- Streaming outputs
- OpenAI-compatible API server
- Support for x86_64, PowerPC CPUs, Arm CPUs and Applie Scilicon (CPU inference). This package does not support any GPU inference. For GPU inference support use the official vLLM PypI
- Prefix caching support
- Multi-LoRA support
vLLM seamlessly supports most popular open-source models on HuggingFace, including:
- Transformer-like LLMs (e.g., Llama)
- Mixture-of-Expert LLMs (e.g., Mixtral, Deepseek-V2 and V3)
- Embedding Models (e.g., E5-Mistral)
- Multi-modal LLMs (e.g., LLaVA)
Find the full list of supported models here.
Importnt Notes
- Install this package on Linux envirenment only. For Windows you will have to use WSL2 or later
- This package has a Container.io (Docker/Podman etc.) compatible image in Docker Hub
- Apache Licence of main vLLM project
- GPL License of this CPU specific vLLM package
Getting Started
Install vLLM with pip or uv
mkdir -p /path/to/vllm
cd /path/to/vllm
uv venv
uv pip install torch==2.8.0 torchvision --index-url https://download.pytorch.org/whl/cpu
uv pip install vllm-cpu-avx512bf16
Install uv on Linux environment using CURL:
curl -LsSf https://astral.sh/uv/install.sh | sh
or using WGET
wget -qO- https://astral.sh/uv/install.sh | sh
if you wanna install a specific version of uv
curl -LsSf https://astral.sh/uv/0.9.11/install.sh | sh
vllm-cpu
This CPU specific vLLM has 5 optimized wheel packages from the upstream vLLM source code:
| Package | Optimizations | Target CPUs |
|---|---|---|
vllm-cpu |
Baseline (no AVX512) | All x86_64 and ARM64 CPUs |
vllm-cpu-avx512 |
AVX512 | Intel Skylake-X and newer |
vllm-cpu-avx512vnni |
AVX512 + VNNI | Intel Cascade Lake and newer |
vllm-cpu-avx512bf16 |
AVX512 + VNNI + BF16 | Intel Cooper Lake and newer |
vllm-cpu-amxbf16 |
AVX512 + VNNI + BF16 + AMX | Intel Sapphire Rapids (4th gen Xeon+) |
Each package is compiled with specific CPU instruction set flags for optimal inference performance.
Check available CPU instruction sets
lscpu | grep -i flags
Example list of CPUs with their supported instruction sets
| CPU Architecture (Intel/AMD) | AVX2 | AVX-512 F (Base) | VNNI (INT8) | BF16 (BFloat16) (via AVX-512) | AMX-BF16 (via Tile Unit) |
|---|---|---|---|---|---|
| Intel 4th Gen / AMD Ryzen Zen2 & Newer | Yes | No | No | No | No |
| Intel Skylake-SP / Skylake-X / AMD Zen 4 & Newer | Yes | Yes | No | No | No |
| Intel Cooper Lake (3rd Gen Xeon) / AMD Zen 4 (EPYC) / Ryzen Zen5 & Newer | Yes | Yes | Yes | Yes | No |
| Intel Sapphire Rapids (4th Gen Xeon) & Newer | Yes | Yes | Yes | Yes | Yes |
***Currently no AMD CPU support AMXBF16. AMD expected to include AMXBF16 support from AMD Zen 7 CPUs
Buy Me a Coffee
Your support encourages me to keep creating/supporting my open-source projects. If you found value in this project, you can buy me a coffee to keep me up all the sleepless nights.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file vllm_cpu_avx512bf16-0.9.2-cp312-cp312-manylinux_2_17_x86_64.whl.
File metadata
- Download URL: vllm_cpu_avx512bf16-0.9.2-cp312-cp312-manylinux_2_17_x86_64.whl
- Upload date:
- Size: 10.1 MB
- Tags: CPython 3.12, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
f6976ba422a53a9438f2c2793ea90562e14967e9710ae5a08c12d82b7766e399
|
|
| MD5 |
0c1c8ae16ac09507954f927a6045b976
|
|
| BLAKE2b-256 |
e82e3abed6825d29dcbc8b98b342aa5790739a6ab1550463d203a4a63d4f3381
|
File details
Details for the file vllm_cpu_avx512bf16-0.9.2-cp311-cp311-manylinux_2_17_x86_64.whl.
File metadata
- Download URL: vllm_cpu_avx512bf16-0.9.2-cp311-cp311-manylinux_2_17_x86_64.whl
- Upload date:
- Size: 10.1 MB
- Tags: CPython 3.11, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
984d1897bce063bd44fd7bd99dc4d1ca3549ae1a80fa8d2642a0c365ca635e48
|
|
| MD5 |
0f488d8ff9959d6f9d06bd57fb600fd8
|
|
| BLAKE2b-256 |
091bdc38279d677a3cd9e38a7923addbf434ab06421a01766ecbf960deecf288
|
File details
Details for the file vllm_cpu_avx512bf16-0.9.2-cp310-cp310-manylinux_2_17_x86_64.whl.
File metadata
- Download URL: vllm_cpu_avx512bf16-0.9.2-cp310-cp310-manylinux_2_17_x86_64.whl
- Upload date:
- Size: 10.1 MB
- Tags: CPython 3.10, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
bca63be8f1edd6dee0b5362a71bfca5e2735e88dcab44b62cea199b9387d23b0
|
|
| MD5 |
ae2bda2ac409edc2d3d81aae79d553e9
|
|
| BLAKE2b-256 |
f1538492bf07277f13bdda90b939811959f602214bf2fa12510d1fa868e97ce7
|
File details
Details for the file vllm_cpu_avx512bf16-0.9.2-cp39-cp39-manylinux_2_17_x86_64.whl.
File metadata
- Download URL: vllm_cpu_avx512bf16-0.9.2-cp39-cp39-manylinux_2_17_x86_64.whl
- Upload date:
- Size: 10.1 MB
- Tags: CPython 3.9, manylinux: glibc 2.17+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0842f092bf02a7dc4f603c5a2d0b0a9dab0e549379477ba5567b249aba86ad44
|
|
| MD5 |
d13444e856a7234605b42e8ba4a6636f
|
|
| BLAKE2b-256 |
9570285c90ae2371967ff90402ab8d8a55ae4b276772014ae0c5f4f967e92ef5
|