jax

Update skip_on_devices and mosaic skip order in gpu_pallas_distributed_Test.py

#669

Merged