Using ROCm for AI

Applies to Linux

2024-06-21

3 min read time

ROCm offers a suite of optimizations for AI workloads from large language models (LLMs) to image and video detection and recognition, life sciences and drug discovery, autonomous driving, robotics, and more. ROCm proudly supports the broader AI software ecosystem, including open frameworks, models, and tools.

For more information, see What is ROCm?

In this guide, you’ll learn about:

• Installing ROCm and machine learning frameworks

• Training a model

• Running models from Hugging Face

• Deploying your model

To learn about ROCm for HPC applications and scientific computing, see Using ROCm for HPC.