Add ROCm support

[xys-syx]

I have not added lit tests yet

[github-actions]

EnzymeJAX Benchmarks

Benchmark suite	Current: 3cd2a39	Previous: 0341e5c	Ratio
scatter_sum / JaX / cpu / Primal	0.000004364811900086351 s	0.000004350405000150204 s	1.00
scatter_sum / JaXPipe / cpu / Primal	0.000004357116900064284 s	0.000004312506999849574 s	1.01
scatter_sum / JaX / tpu / Primal	0.0001315213231999 s	0.0001511090920997 s	0.87
scatter_sum / JaXPipe / tpu / Primal	0.0001314510052001 s	0.0001527826641999 s	0.86

This comment was automatically generated by workflow using github-action-benchmark.

[wsmoses]

Use --sandbox_debug to see verbose messages from the sandbox and retain the sandbox build root for debugging
src/enzyme_ad/jax/Passes/ConvertPolygeistToLLVM.cpp:2217:8: error: unused variable 'ptrty' [-Werror,-Wunused-variable]
 2217 |   auto ptrty = LLVM::LLVMPointerType::get(rewriter.getContext());
      |        ^~~~~
      ````