Jithendaraa Subramanian

I think jax.partial was discontinued from 0.2.21. If you use the same version as that of the authors (0.2.18) you don't really need to shift to functools.

Your error also seems to do with this change from jax.partial to functools.partial. I am trying to reproduce the experiments as well and though I am not fully done, I...

Also, does you `jax` access GPU (you can check this using `jax.devices()`

Hey @lucfra, That is a bit surprising since I see no reason why it would't work. `jax.partial` was removed only in versions `>=0.2.20`. So technically it should work and it...

Here are the values from my scipy and jaxopt runs, in that order Final objective: **4.84871** vs **1.5213378e+20** which seems concerning. Optimized value of `wa_new` (scipy/numpy) ``` array([0.00000000e+00, 1.58362135e+00, 0.00000000e+00,...