Skip to main content

Ctrl+K

ROCm™ AI Ecosystem

Core SDK
AI Ecosystem
GPU Systems and Infrastructure
Toolkits

Blogs
Developer Hub

AMD ROCm AI Ecosystem

ROCm compatibility matrix

Deep learning frameworks

Install PyTorch
Install JAX

Inference

vLLM
SGLang
MIGraphX
ONNX Runtime
xDiT diffusion inference
ComfyUI image generation

Distributed inference

vLLM with MoRI recipe
SGLang with MoRI recipe

Training

Training with Primus and Megatron
Training with Primus and PyTorch
Training with Primus and JAX MaxText
Scale model training

Optimization

Workload optimization
vLLM V1 performance optimization
Model quantization techniques
Model acceleration libraries
Optimize Triton kernels
Optimize with Composable Kernel

Tutorials

AI Playbooks
AI Developer Hub

Reference

System setup for AI workloads

Search

Ctrl+K

Terms and Conditions
ROCm Licenses and Disclaimers
Privacy
Trademarks
Supply Chain Transparency
Fair and Open Competition
UK Tax Strategy
Cookie Policy
Cookie Settings

© 2026 Advanced Micro Devices, Inc