Hacker News Reader

How about Pytorch?

You have to compile it yourself, which sucks. For some reason PyTorch aren't agreeing to upstream ROCm, yet.

You don't need to compile it, AMD provides docker images.

For example I was able to run fast.ai/pytorch oh my amd gpu https://github.com/briangorman/fastai_rocm_docker