But now we need to wait for PyTorch ROCm 6.0…
This is the best summary I could come up with:
Earlier this month at AMD’s AI event in San Francisco they announced ROCm 6.0 while launching the MI300X and MI300A accelerators.
ROCm 6.0 also brings support for more AI libraries including DeepSpeed, ONNX-RT, Jax, CuPy and others.
There is also now FP8 support in PyTorch, hipSPARSELt to provide sparse matrix core technique support on RDNA/CDNA GPUs for speeding up AI workloads and inferencing, and improved documentation.
ROCm 6.0 additionally now puts the AMD instinct MI50 on end-of-support notice.
We also don’t have any better information to communicate right now on when the rest of the AMD Radeon RX 7000 series line-up may see official ROCm support.
Lastly the Instinct MI300, MI250, MI210, and MI100 series are fully supported by ROCm 6.0 while MI50 is reaching end-of-support.
The original article contains 355 words, the summary contains 126 words. Saved 65%. I’m a bot and I’m open source!
deleted by creator