Alexander Grund

> OMP_NUM_THREADS=2 ? That doesn't affect the thread pools created by jax/xla. I found `PJRT_NPROC` for that but setting `PJRT_NPROC=32` in `local_test_exports` also failed. Currently experimenting with both... @verdurin How...

Ok, maybe it isn't the number of threads after all. I tried with PJRT_NPROC=2 to only create a small number of threads, well below the ones on the working 64/20...

I found that there is a difference when running the tests on a machine with or without GPUs. I have a 96 core machine with GPUs and the build succeeds....