Skip to content

Jax tests for zn search#129

Open
temp-noob wants to merge 3 commits intoStingraySoftware:mainfrom
temp-noob:jitTestsForZnSearch
Open

Jax tests for zn search#129
temp-noob wants to merge 3 commits intoStingraySoftware:mainfrom
temp-noob:jitTestsForZnSearch

Conversation

@temp-noob
Copy link
Copy Markdown

@temp-noob temp-noob commented Apr 3, 2026

Adding tests as to demonstrate the utility of Jax powered Zn Search.
Following are the results for CPU (enforced by the env variable JAX_PLATFORM=cpu):
Screenshot 2026-04-02 at 10 32 03 PM
Following are the results for GPU (my laptop is macbook M5 pro)

Screenshot 2026-04-02 at 10 36 36 PM The command used to run the notebook is `ENABLE_PJRT_COMPATIBILITY=1 jupyter nbconvert \ --to notebook --execute --inplace \ --ExecutePreprocessor.timeout=1800 \ "notebooks/Pulsar/Pulsar search with epoch folding and Z squared.ipynb"`

This change is Reviewable

Mohit Tripathi and others added 3 commits March 25, 2026 22:27
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
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.

1 participant