Torch fx and distributed training

torch.compile FX (Functional Transformations)
1

Can I use torch.fx to generate a distributed training program by inserting communication operations?

1 Like