Optimized for AI Workloads

AMD ROCm™ is an open software stack offering a suite of optimizations for AI workloads and supporting the broader AI software ecosystem including open frameworks, models, and tools.

Simplified Model Development

Supports a wide range of models on Hugging Face that can be leveraged to develop user-specific solutions

Support for Leading Frameworks

Provides upstream support for leading AI frameworks including TensorFlow and PyTorch

Machine Learning Libraries

Offers dedicated libraries for machine learning, including MIOpen and MIVisionX

Scale-Out Deployments

Supports leading containerization tools including Docker, Singularity, Kubernetes®, and Slurm to enable deployment at scale

Latest News

Read about the latest advancements in AI development using ROCm.

Meta logo on background

LLAMA 3

LLAMA 3—the Meta next-generation, state-of-the-art, open source LLM—runs out-of-the-box on AMD Instinct™ accelerators with the support of ROCm.

Read Blog
OpenAI logo

GPT-4

The most advanced model yet from OpenAI—GPT-4—runs on AMD Instinct accelerators with the support of ROCm.

Read News
Hugging Face logo

Hugging Face

ROCm supports thousands of models on Hugging Face that can be leveraged to develop user-specific solutions.

Learn More

ROCm for AI – Open Software Stack

AI Solutions and Services

  • Computer Programming
  • Content Generation
  • Customer Chatbots
  • Data Analysis
  • Developer Tools
  • Employee Copilots
  • Knowledge Management
  • Process Automation
  • Recommendation Systems
  • Report Generation
  • Summarization

  • Bloom
  • Falcon LLM
  • GPT-4
  • Llama 2
  • Llama 3
  • Mistral AI
  • Stable Diffusion
  • StarCoder
  • Qwen
  • Yi

AI Ecosystem

ROCm AI partner logos

GPU

AMD Instinct white logo
AMD Radeon white logo
Programming code abstract technology background of software developer and Computer script

Access the Latest AI Models

The AMD Infinity Hub contains a collection of advanced software containers and deployment guides for AI applications.

Access Infinity Hub

Build your AI Models on the AMD Developer Cloud

Apply for access to AMD Instinct™ accelerators and streamline your development.

Apply Today
AMD Data Center and Cloud Computing

Partnerships and Proven Success

See how enterprises and leading research institutes have partnered with AMD to advance AI.

Lamini logo

Lamini

Discover how Lamini software and ROCm provide a solid foundation for high-performance applications like finetuning LLMs.

Read Blog
Mosaic ML logo

Mosaic ML

See how Mosaic ML leverages ROCm to enable LLM training on AMD Instinct accelerators with zero code changes and at high performance.

Read Blog
Hugging Face logo

Hugging Face

Learn about how Hugging Face and AMD partner to accelerate state-of-the-art models, and how Hugging Face is integrating ROCm in their open-source libraries.

Read Blog
Pytorch logo

PyTorch

AMD is aligning with the PyTorch Foundation on the value of an open software ecosystem for AI and is driving ROCm support for PyTorch.

Read Blog

Supported GPUs

Developer Resources

ROCm Developer screen shot

ROCm Developer Hub

Start developing AMD GPU-accelerated applications. Visit the ROCm Developer Hub to get access to the latest user guides, containers, training videos, webinars, and more.

Get Started

ML Development on AMD Radeon GPUs

Researchers and developers working with ML models and algorithms using PyTorch can now also use ROCm (starting with version 5.7) on Ubuntu® Linux® to tap into the parallel computing power of the latest AMD Radeon 7900 series desktop GPUs.²

Get Started

ROCm Webinars

Watch past webinars on register for upcoming ones.

Watch Now

ROCm Newsletter

Receive the latest ROCm news.

Stay Informed
Footnotes

©2024 Advanced Micro Devices, Inc. All rights reserved. AMD, the AMD Arrow logo, AMD ROCm, AMD Instinct, EPYC, Radeon Instinct, and combinations thereof are trademarks of Advanced Micro Devices, Inc. PyTorch is a trademark or registered trademark of PyTorch. Other product names used in this publication are for identification purposes only and may be trademarks of their respective companies.

  1. For a full list of Radeon parts supported by ROCm, go to https://rocm.docs.amd.com/en/latest/reference/gpu-arch-specs.html
  2. Radeon™ AI technology is compatible with all AMD Radeon 7000 Series graphics cards and newer. Please check with your system manufacturer for feature availability prior to purchase. GD-232.