Skip to content

Feat: proximal module#200

Draft
mrava87 wants to merge 10 commits into
PyLops:mainfrom
mrava87:feat-prox
Draft

Feat: proximal module#200
mrava87 wants to merge 10 commits into
PyLops:mainfrom
mrava87:feat-prox

Conversation

@mrava87

@mrava87 mrava87 commented Jun 19, 2026

Copy link
Copy Markdown
Contributor

This PR is a starting point of an effort to support distributed proximal operators and solvers from pyproximal in pylops-mpi

To begin with the following operators/solvers are supported:

  • any separable proximal operator (i.e., where prox is taken element-wise) via simple wrapping on the pyproximal operator into a MPIProxOperator
  • A custom distribured proximal operator for L2 norm named MPIL2 which leverages distributed stacking and solvers to solve augumented systems of equations in the prox operator.
  • ProximalGradient and ADMML2 solvers

These already enable the solution of 2 typical use-cases of convex optimization:

  • sparsity promoting inversion: added to reflectivity tutorial
  • TV-reg inversion: added to poststack tutorial
    Note that for this to be possible, I needed to extendStackedDistributedArrayto operate with nestedStackedDistributedArray`. This will be handled separately in Feat: allow nested stacked arrays #201 and then merged back into here.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants