jax

0bd8fca8 - [pallas] `pallas_call` no longer supports multiple `compiler_params=`

[pallas] `pallas_call` no longer supports multiple `compiler_params=` This feature turned out to be unhelpful in practice, since we almost never have real-world kernels sharing the same implementation for multiple different backends. I decided not to go through a deprecation cycle as I expect that nobody (or very very few users) actually do `compiler_params={`. Reverts 3d5fc32f610b1f939deb42dc0eb5b7cd27df2cd7 PiperOrigin-RevId: 868057783