Olli Lupton

There are several pytests that fail when run against the latest JAX etc. See for example https://github.com/NVIDIA/JAX-Toolbox/actions/runs/14589659816/job/40924621016#step:7:562 ``` =================================== FAILURES =================================== _________________________________ test_module __________________________________ def test_module(): denses_features_replaces = [] def...

//tests/mosaic:gpu_test_distributed_gpu and //tests/pallas:gpu_pallas_distributed_test_gpu fail on single-GPU machines

1

### Description These are tagged as `multiaccelerator` tests, but I understand that they are still expected to pass/skip on machines that only have a single GPU attached or visible. For...