openfl
openfl copied to clipboard
Add support for PyTorch 2.0+ in OpenFL
This task involves adding support for PyTorch 2.0+ in the existing PyTorchTaskRunner. Certain models during the forward pass may fail due to major changes in the torch.compile(...) method.