jax

73fa0f48 - [Pallas] Deprecate dictionary compiler_params in favor of dataclass.