pytorch-operator
pytorch-operator copied to clipboard
[feat] Support PyTorch 1.9
Ref https://github.com/pytorch/elastic/issues/117
@gaocegege Any plans on your side?
I am investigating 1.9.0, maybe we can support it when 1.9.1 is released. The feature is not stable now. WDYT
I will write a design proposal for it, based on the unified operator.