[Question] Can those matrix inversion linear solver with neumann series approximation be speed up by GPU?

[happpyosu]

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