New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

fix lasso with scalar l1reg #604

Open

BalzaniEdoardo wants to merge 1 commit into google:main from BalzaniEdoardo:fix_prox_lasso

BalzaniEdoardo commented Jul 23, 2024

First of all, thanks for the very nice package.

In this PR I am generalizing the behavior of jaxopt.prox.prox_lasso to handle scalar input when jit-compiled.

In particular the following used to result in an exception.

import numpy as np
from jaxopt import prox
import jax

rng = np.random.RandomState(0)
x = (rng.rand(20) * 2 - 1, rng.rand(20) * 2 - 1)
jax.jit(prox.prox_lasso)(x, 0.5)

Best regards,
Edoardo


          fix lasso with scalar l1reg

9b9ccd5

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet