jax

bb6ca370 - [pallas] Use `*Params` dataclasses to pass `compiler_params=` to `pl.pallas_call`

[pallas] Use `*Params` dataclasses to pass `compiler_params=` to `pl.pallas_call` Dataclasses provide better ergonomics and also improve type safety, making it easier for Pallas maintainers to change the structure of `compiler_params=`. PiperOrigin-RevId: 748266655