jax

2c23d5df - [Pallas:SC] Properly select the serialized IR version to avoid libtpu incompatibility