jaxopt.linear_solve.solve_bicgstab

jaxopt.linear_solve.solve_bicgstab(matvec, b, ridge=None, init=None, **kwargs)[source]

Solves A x = b using bicgstab.

Parameters
  • matvec (Callable) – product between A and a vector.

  • b (Any) – pytree.

  • ridge (Optional[float]) – optional ridge regularization.

  • init (Optional[Any]) – optional initialization to be used by bicgstab.

  • **kwargs – additional keyword arguments for solver.

Return type

Any

Returns

pytree with same structure as b.