xla

Torchax: Allow reuse of the jittable procedure with different functio…

#9374

Merged