JAXOPT Projected Gradient

[pulkitchhabra19]

Hello, I have been using JAXOPT Projected Gradient approach with projection box constraints. And I would like to get more output from the algorithm than it is providing such as I would like to have to loss value at each iteration as output. I believe I can set verbose to False to display the error at each iteration but instead I'd like it as output.

Secondly, I believe the tolerance that is passed to the Proximal Gradient class is not used anywhere and so if that is the case, this might be a bug as no matter what tolerance I set, it would not matter. Could you please look into it? Thanks.

Irrespective of what tolerance I put, it runs to the max iterations and so I looked at this and found that tolerance is not used at all:

jaxopt/jaxopt/_src/proximal_gradient.py

Line 153 in 501cc20

tol: float = 1e-3