Delay using tuple arguments in Shardonnay until after it's run.
Delay using tuple arguments in Shardonnay until after it's run.

Users of XLA can choose whether XLA should use tuple args for the params. This is a problem as when arguments become a tuple they lose their frontend attributes, which is where we keep our `xla.sdy.sharding` attributes. So we can't create tuple arguments until after Shardonnay is run.

Code that doesn't relay on Shardonnay behaves the same way with using tuple args all the way through.

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/13639 from philipphack:u_layer_uaf_xla 91ebf7b4a2ac90ebadce27d1a73e88fb4513aed4
