[Question] Can those matrix inversion linear solver with neumann series approximation be speed up by GPU? #231
Description
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
Thank you for your brilliant work fisrtly. Currently, I am using your framework to solve a bi-level optimization problem with implict diff. I am using matrix inversion linear solver with neumann series approximation, however, I find this process is slow when dealing with large-scale problem. I just wonder if matrix inversion with linear solvers can be accerlated by using GPUs? Thanks