jaxopt.prox.prox_non_negative_ridge

jaxopt.prox.prox_non_negative_ridge(x, l2reg=1.0, scaling=1.0)

Proximal operator for the squared l2 norm on the non-negative orthant.

$$\underset{y\ge 0}{\text{argmin}}\frac{1}{2}||x-y|{|}_2^2+\text{scaling}\cdot \text{l2reg}\cdot ||y|{|}_2^2$$

Parameters

• x (Any) – input pytree.

• l2reg (Optional[float]) – regularization strength.

• scaling (float) – a scaling factor.

Returns

output pytree, with the same structure as x.