[Question] TorchScirpt for ImplicitMetaGradientModule

[ChiahsinChu]

Required prerequisites

• I have read the documentation https://torchopt.readthedocs.io.
• I have searched the Issue Tracker and Discussions that this hasn't already been reported. (+1 or comment there if it has.)
• Consider asking first in a Discussion.

Questions

Hi,

I want to implement a method with implicit gradient solver. I therefore check the examples in https://github.com/metaopt/torchopt/blob/main/tutorials/5_Implicit_Differentiation.ipynb. However, the OOP API seems not compatible with torchscript. I am wondering if there is a way to overcome this issue?

Thanks in advance.