jax

36cfce15 - Merge pull request #38946 from aybchan:aybchan/update-thor-gpuinfo-test