jax

cd5f6566 - [pallas:mgpu] Removed an unnecessary test for `pl.pallas_call` monkey patching