`apply` function in custom extension not supported by TorchScript

jit
1

Hello! I’m trying to run compile my model that contains a custom op with TorchScript using the method in the following link:
https://pytorch.org/tutorials/advanced/torch-script-parallelism.html

However, it failed with the following error

Python builtin <built-in method apply of FunctionMeta object at 0x55ad125f0e18> is currently not supported in TorchScript.

indicating the apply function in the custom extension code is not supported by TorchScript. I wonder if there’s any way to fix this? Also is there any plan to support apply in TorchScript recently?

Thanks in advance!