Jax tests for zn search#129

Open

temp-noob wants to merge 3 commits intoStingraySoftware:mainfrom

temp-noob:jitTestsForZnSearch

Commits on Mar 26, 2026

notebooks: compare standard and JAX z_n_search

authored and
Mohit Tripathi
committed
notebooks: add fdot JAX benchmark and backend check

authored and
Mohit Tripathi
committed

Commits on Apr 3, 2026

jit tests for zn search. Also add tests for 2D search

Mohit Tripathi
authored and
Mohit Tripathi
committed