PLS, a paper related question I want to ask

[KylinC]

I want to know how the activations (or jax_op data) transfer between different stage(or submesh), it's done by ray or jaxlib automatically once we get the executable? or we write some code for the activation transfer? (if it is, I want to know where it is)
Looking forward your reply, Thx

[merrymercy]

We wrote some code for it with nccl