Article Complete
AIM
AMD Extends PyTorch Support on RDNA-3 For Local ML Training

AMD Extends PyTorch Support on RDNA-3 For Local ML Training

AMD is also trying to break NVIDIA's CUDA monopoly in the AI parallel computing segment.

AMD Extends PyTorch Support on RDNA-3 For Local ML Training
Mohit Pandey·Published Updated

Key takeaways

  • AMD extends PyTorch support on RDNA 3 for enhanced local ML training capabilities.
  • Leverage AMD ROCm 5.7 to utilize parallel computing on Radeon RX 7900 XTX GPUs.
  • Achieve up to 2X higher AI performance per compute unit compared to previous generations.
  • Reduce reliance on cloud solutions with cost-effective local ML training options.
  • AMD aims to challenge NVIDIA's CUDA dominance in the AI parallel computing market.
5 min

In a move aimed at enhancing accessibility to AI for developers and researchers, AMD had unveiled ROCm 5.7 last month, the latest iteration of its open software ecosystem for accelerated computing. Now,  the company has now extended its support for PyTorch machine learning (ML) to AMD RDNA 3-based graphics via ROCm 5.7.

This means that individuals engaged in working with ML models and algorithms in PyTorch can leverage AMD ROCm 5.7 on Ubuntu Linux. This enables them to harness the parallel computing capabilities offered by AMD Radeon RX 7900 XTX and Radeon PRO W7900 GPUs, which come equipped with 192 dedicated AI accelerators. 

These accelerators promise up to 2X higher AI performance per compute unit in comparison to the preceding generation. The integration of ML on desktops provides users with a local, private, and cost-effective avenue for supporting ML training and inference, thereby reducing the reliance on cloud-based solutions.

Read: AMD’s Attempt to Break NVIDIA’s CUDA

AMD is also trying to break NVIDIA's CUDA monopoly in the AI parallel computing segment. AMD has made a huge leap with its recent bet on Nod.ai, an open source AI software firm. Nod.ai has been known for developing a portfolio of tools and systems for boosting AI applications on AMD hardware. 

The software bet has been going on at AMD for some time now. In August, the company also announced the acquisition of Mipsology, a French AI startup, which has also been a long-standing AMD partner and developing AI software for the chipmaker, similar to Nod.ai.

Membership

Stay ahead with AI insights.

Unlimited access to premium articles, exclusive analysis, and the stories shaping India's AI economy.

Subscribe now

AIM Americas

1190 Miraloma Way Suite P, Sunnyvale, CA 94085

AIM India

1st Floor, Sakti Statesman, Bellandur, Bengaluru – 560103

© 2026 AIM. All rights reserved.