search

lanl / scico

Scientific Computational Imaging COde
BSD 3-Clause "New" or "Revised" License
90 stars 17 forks source link

Add diagonal operator mapping base operator over an array axis #521

Closed bwohlberg closed 1 week ago

bwohlberg commented 1 month ago

Add diagonal operator mapping base operator over an array axis via jax.pmap or jax.vmap.

Resolves #413.