jax

4b9bbb5a - [JAX] Populate Send/Recv frontend attributes.

[JAX] Populate Send/Recv frontend attributes. Populate the frontend attributes to target PjRt Rendezvous. This allows for functions with send/recv operations to be used from multiple callsites. PiperOrigin-RevId: 856253455