Kubernetes GPU virtualization and heterogeneous accelerator scheduling for AI infrastructure.
HAMi stands for Heterogeneous AI Computing Virtualization Middleware. Formerly known as k8s-vGPU-scheduler, HAMi helps platform teams share expensive GPUs and other AI accelerators across Kubernetes workloads, isolate device memory and compute, and schedule pods with device-aware policies without changing application code.
HAMi is a CNCF Sandbox and CNCF Landscape project. It is also listed in the CNAI Landscape.
AI infrastructure teams often run into the same Kubernetes accelerator problems: whole GPUs are allocated to small jobs, teams compete for scarce devices, different accelerator vendors expose different operational models, and schedulers lack enough device context to place workloads efficiently.
HAMi provides a Kubernetes-native layer for:
HAMi is composed of a mutating webhook, scheduler extender, device plugins, and device-specific in-container virtualization components.
Pod submission
-> HAMi mutating webhook
-> HAMi scheduler filter / score / bind
-> device allocation written to pod annotations
-> device plugin Allocate()
-> container runtime environment
-> HAMi monitor and metrics
HAMi lets workloads request only the accelerator resources they need. For example, the following pod asks for one physical NVIDIA GPU with 3 GiB of GPU memory:
resources:
limits:
nvidia.com/gpu: 1
nvidia.com/gpumem: 3000
The workload sees the allocated device resources inside the container, while HAMi coordinates scheduling, allocation, and isolation.
Notes:
- After installing HAMi, the value of
nvidia.com/gpuregistered on the node defaults to the number of vGPUs.- When requesting resources in a pod,
nvidia.com/gpurefers to the number of physical GPUs required by the current pod.
HAMi supports multiple heterogeneous accelerator backends, including GPUs, NPUs, DCUs, MLUs, GCUs, XPUs, and more. Device capabilities vary by vendor, model, driver, and hardware generation.
See the current HAMi supported devices page for the maintained support matrix.
For the NVIDIA device plugin path, prepare:
nvidia-docker version > 2.0Label GPU nodes so HAMi can manage them:
kubectl label nodes <node-name> gpu=on
Add the HAMi Helm repository:
helm repo add hami-charts https://project-hami.github.io/HAMi/
helm repo update
Install HAMi:
helm install hami hami-charts/hami -n kube-system
Verify that the scheduler and device plugin are running:
kubectl get pods -n kube-system
When hami-device-plugin and hami-scheduler are both Running, submit an example workload:
kubectl apply -f examples/nvidia/default_use.yaml
For the complete installation guide and configuration options, see the HAMi documentation.
HAMi supports multiple scheduling modes for AI workloads:
HAMi works with the default Kubernetes scheduler path and can also be used with Volcano for batch-oriented AI workloads. See the HAMi website for current scheduler integration guides.
HAMi exposes metrics for monitoring cluster accelerator usage. After installation, metrics are available through the scheduler monitor endpoint:
http://<scheduler-ip>:<monitor-port>/metrics
The default monitor port is 31993. You can change it with Helm values such as --set scheduler.service.monitorPort=<port>.
HAMi also provides:
| Project | What the integration enables |
|---|---|
| vLLM | Run inference servers with GPU memory caps, enabling multiple models to share one GPU |
| Volcano | Gang scheduling and queue-based batch scheduling for GPU workloads |
| Kueue | HAMi resources exposed to Kueue via ResourceTransformation for batch job queueing |
| Prometheus | HAMi exposes per-container GPU metrics including memory usage and utilization |
| Grafana | Pre-built dashboard available for visualizing HAMi GPU metrics |
| NVIDIA GPU Operator | Can coexist with GPU Operator when HAMi manages scheduling and the Operator manages drivers |
HAMi is governed by maintainers and contributors. Governance is described in the HAMi community repository.
To contribute code, documentation, tests, or device backend improvements, read CONTRIBUTING.md.
The HAMi community is open to users, contributors, hardware vendors, and platform teams building Kubernetes-based AI infrastructure.
| Event | Talk |
|---|---|
| CHINA CLOUD COMPUTING INFRASTRUCTURE DEVELOPER CONFERENCE, Beijing 2024 | Unlocking heterogeneous AI infrastructure on k8s clusters |
| KubeDay Japan 2024 | Unlocking Heterogeneous AI Infrastructure K8s Cluster: Leveraging the Power of HAMi |
| KubeCon + AI_dev Open Source GenAI & ML Summit China 2024 | Is Your GPU Really Working Efficiently in the Data Center? N Ways to Improve GPU Usage |
| KubeCon + AI_dev Open Source GenAI & ML Summit China 2024 | Unlocking Heterogeneous AI Infrastructure K8s Cluster |
| KubeCon Europe 2024 | Cloud Native Batch Computing with Volcano: Updates and Future |
HAMi is licensed under the Apache License 2.0. See LICENSE for details.
Copyright Contributors to HAMi, established as HAMi a Series of LF Projects, LLC.