Skip to content

Support JAX backend in ChirpRadon2D/3D by removing in-place assignment.#681

Merged
mrava87 merged 3 commits into
PyLops:devfrom
GeophyAI:fix_jax
Jul 8, 2025
Merged

Support JAX backend in ChirpRadon2D/3D by removing in-place assignment.#681
mrava87 merged 3 commits into
PyLops:devfrom
GeophyAI:fix_jax

Conversation

@GeophyAI
Copy link
Copy Markdown
Contributor

@GeophyAI GeophyAI commented Jul 8, 2025

This PR replaces in-place assignments in ChirpRadon2D with pylops.backend.inplace_set, enabling compatibility with JAX arrays.

@mrava87 mrava87 changed the title Support JAX backend in ChirpRadon2D by removing in-place assignment. Support JAX backend in ChirpRadon2D/3D by removing in-place assignment. Jul 8, 2025
@mrava87
Copy link
Copy Markdown
Collaborator

mrava87 commented Jul 8, 2025

Thanks @GeophyAI

I have just pushed a similar change for ChirpRadon3D and checked they both work and produce same results to the numpy backend. Happy to merge this 😄

@mrava87 mrava87 merged commit 7837aa8 into PyLops:dev Jul 8, 2025
14 checks passed
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