Cool paper from my former colleagues @markboss.bsky.social and Varun Jampani!

Thanks to @frame.work for providing a pre production framework desktop for running and developing the Windows support for PyTorch+rocm. It’s been a great experience so far.

The 16 cores of Strix halo helped a lot with compilation times, and PyTorch runs nicely too!

This is a "fat" wheel which bundles ROCm binaries alongside pytorch, so you won't need to download/install ROCm yourself.

You can get even higher performance when using “—gpu-only” - give it a try!

Release Pytorch wheels for gfx110x, gfx1151, and gfx1201 · scottt/rocm-TheRock Self-contained Pytorch wheels for gfx1100, gfx1101, gfx1102, gfx1103, gfx1151, and gfx1201 GPUs for Python 3.12 on Windows by @jammm Mainly tested and known to run fast on the Strix Halo (gfx1151)...

Python312 Wheel can be downloaded from
github.com/scottt/rocm-...
Currently tested to work fast for gfx1151 but It should technically support gfx1100,gfx110{1,2,3}, gfx1151 gfx1201, gfx1151. Basically 7900(XT)/7800(XT)/7600(XT)/phoenix, 9070(XT) (navi48) and and strix halo (8060s).

comfyui running out of the box nicely AMD Strix Halo on windows+ROCm.
10.5 it/s on the default sd1.5 workflow with `python main.py --use-pytorch-cross-attention`

Wheels available - made with @scotttsai.bsky.social - see next post

Stable Virtual Camera: Generative View Synthesis with Diffusion Models

Jensen (Jinghao)Zhou, Hang Gao, Vikram Voleti, @adyaman.bsky.social, Chun-Han Yao, @markboss.bsky.social, @philiptorr.bsky.social, Christian Rupprecht, Varun Jampani

arxiv.org/abs/2503.14489