Jeremiah Willcock

OSQP solvers with fun set and no init_params have misleading error message

8

The error message for setting `fun` to a function and not providing `init_params` to a class such as `jaxopt.BoxOSQP` is incorrect: ``` ValueError: init_x must be provided when fun is...

Type precision issue in BoxOSQP

10

When I have `float64` support enabled in JAX and try to run: ``` optimizer = jaxopt.BoxOSQP() optimizer.run( params_obj=( jnp.eye(30, dtype=jax.numpy.float32), jnp.ones((30,), dtype=jax.numpy.float32), ), params_eq=jnp.ones((1, 30), dtype=jax.numpy.float32), params_ineq=(-1, 1), ) ```...

BoxOSQP does not work without equality constraints

5

Trying to run: ``` optimizer = jaxopt.BoxOSQP() optimizer.run( params_obj=( jnp.eye(30, dtype=jnp.float32), jnp.ones((30,), dtype=jnp.float32), ), params_ineq=(-1, 1), ) ``` produces an error (some parts redacted): ``` [.../jaxopt/_src/osqp.py](...) in run(self, init_params, params_obj,...

Strided indexing turns into gather

3

### Description Syntax such as `a[1::100]` where `a` is a JAX tensor inside a JIT appears to turn into a gather operation rather than a strided slice, at least on...