jax

2807ae4e - [Pallas] Fix ()-shaped vectors being materialized in Pallas lowering.

[Pallas] Fix ()-shaped vectors being materialized in Pallas lowering. This fixes some non-intuitive errors where scalar-shaped values in VREGs were being used in operations that expected SREGs. PiperOrigin-RevId: 746146037