feat: Add JAX acceleration support to z_n_search#966

temp-noob wants to merge 3 commits intoStingraySoftware:mainStingraySoftware/stingray:mainfrom

temp-noob:useJaxFor_z_n_searchtemp-noob/stingray:useJaxFor_z_n_searchCopy head branch name to clipboard