GPT-NeoX implements model parallel autoregressive transformers on GPUs using Megatron and DeepSpeed libraries.
DeepSpeed-MII
MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.

About DeepSpeed-MII
DeepSpeed-MII is an open-source Python library for model inference, powered by DeepSpeed-Inference. It allows non-persistent and persistent deployments in a few lines of code and automatically applies optimizations based on model architecture, model size, batch size, and available hardware.
It includes blocked KV caching, continuous batching, Dynamic SplitFuse, and tensor parallelism for text generation with models such as Llama-2, Mixtral, Phi-2, Qwen, Falcon, and OPT. Current releases use Hugging Face for model weights and tokenizers and are limited to listed families; legacy APIs cover additional models including BERT, RoBERTa, Stable Diffusion, Bloom, and GPT-J.
Key features
- Blocked KV caching for text generation inference
- Continuous batching for text generation workloads
- Dynamic SplitFuse for generation scheduling
- Tensor parallelism for multi-GPU inference
- Non-persistent and persistent deployments in code
- Text-to-image inference support for Stable Diffusion
- Open-source Python library for inference deployments
- Automatically selects optimizations based on model and hardware resources
- Documents support for over 37,000 models across listed architectures
- Current release limited to listed model families
- Uses Hugging Face backend for model weights and tokenizers
DeepSpeed-MII at a glance
- Type
- Open-source project
- Pricing
- Not confirmed. Check the site
- Open source
- Yes · Source code
- Language
- English
- Last checked
- Oct 5, 2026 · Online when checked
- Listed
- Oct 5, 2026
DeepSpeed-MII alternatives
Tools and sites you could use instead, ranked by how closely they match.
Extensible Toolkit for Finetuning and Inference of Large Foundation Models.
The colibri-core NLP tool and library works with n-grams, skipgrams, and other linguistic patterns.
Gymnasium provides a standard API for single agent reinforcement learning environments and related utilities.
DEAP provides a Python framework for prototyping evolutionary computation algorithms and testing new ideas.
DyNet provides tools for building dynamic neural networks and deep learning models.
Popular in AI Infrastructure & MLOps
Well-known listings from the same topic.
DeepSpeed-MII FAQ
How does MII work?
Which models are supported?
Does MII support models other than LLMs?
Tell us and an editor will check it.
Claim the listing and suggest updates yourself. An editor checks every claim first.












