jax

184e7bf4 - Lower `jnp.empty` & `jnp.empty_like` to `lax.empty`

Lower `jnp.empty` & `jnp.empty_like` to `lax.empty` This is a breaking change: it means that `jnp.empty` and `jnp.empty_like` now return an uninitialized array rather than an array initialized with zeros. The reason for the prior behavior is that until recently, XLA lacked an instruction for creating unititialized buffers. Reverts c36456a817cc6fb523f238a742d81e1d7a269f8c PiperOrigin-RevId: 938071888