Skip to content

Initial Python implementation of SSP2#4

Open
romanodev wants to merge 6 commits intomainfrom
ssp2_py
Open

Initial Python implementation of SSP2#4
romanodev wants to merge 6 commits intomainfrom
ssp2_py

Conversation

@romanodev
Copy link
Copy Markdown
Collaborator

Description

This PR introduces an initial Python implementation of SSP2.

Known limitations

  • Periodic boundary conditions: Not yet supported. Further discussion is needed regarding endpoint handling.
  • JAX pinned to 0.7.2: Newer versions throw the following error when combining FFT-based filtering with interpax interpolation:
  RET_CHECK failure (external/xla/xla/backends/cpu/runtime/fft_thunk.cc:167)

Example and test: Currently copied from the SSP1 counterparts with just the projection method switched.

@stevengj
Copy link
Copy Markdown
Contributor

stevengj commented May 7, 2026

Is this ready for merging? cc @lxvm

@romanodev
Copy link
Copy Markdown
Collaborator Author

Yes from my side.

There are still several things to be improved upon, such creating a unified test suite for SSP1 and SSP2 (right now they are duplicated), and comparing Python against Julia implementation. However, this initial implementation is self-contained and can be used right away.

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