[Suggestion] torch.func in Pytorch 2.1

leopard-ai / betty

Betty: an automatic differentiation library for generalized meta-learning and multilevel optimization

https://leopard-ai.github.io/betty/

Apache License 2.0

329 stars 27 forks source link

[Suggestion] torch.func in Pytorch 2.1 #19

Open ZeenSong opened 10 months ago

ZeenSong commented 10 months ago

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

sangkeun00 commented 10 months ago

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