Any advice on getting alphafold 2 to work on amd?

I've been trying to get alphafold 2 or more specifically localcolabfold 2 working on my personal computer with an amd gpu. I've been trying for like the past 8 hours with no luck. It can recognise my cpu fine but won't use my gpu no matter what I do. I've tired installing rocm and the rocm jax version both outside and inside of the installation. Every time it either cannot recognise/find the specific jax rocm plugin or there is some dependency issue due to a mismatch of dependencies versions used by rocm jax and normal jax and won't even launch. I've tired everything at this point and I'm sure there is something simple I'm missing due to being super new to all this and not really having much if any background when it comes to python or even Linux stuff in general. I just need some other perspective or something.