jax

4ff78e6a - Remove various methods from `MeshExecutable`

Remove various methods from `MeshExecutable` These are thin and their implementations can be inlined directly at call sites in `XlaExecutable`. Co-authored-by: Roy Frostig <frostig@google.com> PiperOrigin-RevId: 746716734