[Suggestion] torch.func in Pytorch 2.1

[ZeenSong]

Have you considered changing the implementation of IterativeProblem from functorch to torch.func() since the functorch APIs will be deprecated in future versions?

[sangkeun00]

Hello @ZeenSong,

Thanks for the suggestion. When I worked on IterativeProblem, functorch was somewhat unstable, throwing a lot of errors. I am currently working on another project, but will try to revisit this issue once I have more time. In the meantime, feel free to submit PR, if you have a chance to get to work on it.

Best, Sang