fix: better init of spergel HLR and tabulate xValue

.github/workflows/benchmarks.yaml

@@ -40,6 +40,14 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install .[dev]
 
+      # the glibc debug symbols are needed by the benchmark tooling from codspeed
+      - name: Install glibc debug symbols
+        env:
+          DEBIAN_FRONTEND: noninteractive
+        run: |
+          sudo apt-get update
+          sudo apt-get install -y libc6-dbg
+
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v4
         with:

jax_galsim/core/utils.py

@@ -1,6 +1,5 @@
 import re
 import textwrap
-from functools import partial
 from typing import NamedTuple
 
 import equinox
@@ -168,43 +167,6 @@ def ensure_hashable(v):
         return _convert_to_numpy_nan(v)
 
 
-@partial(jax.jit, static_argnames=("niter",))
-def bisect_for_root(func, low, high, niter=75):
-    def _func(i, args):
-        func, low, flow, high, fhigh = args
-        mid = (low + high) / 2.0
-        fmid = func(mid)
-        return jax.lax.cond(
-            fmid * fhigh < 0,
-            lambda func, low, flow, mid, fmid, high, fhigh: (
-                func,
-                mid,
-                fmid,
-                high,
-                fhigh,
-            ),
-            lambda func, low, flow, mid, fmid, high, fhigh: (
-                func,
-                low,
-                flow,
-                mid,
-                fmid,
-            ),
-            func,
-            low,
-            flow,
-            mid,
-            fmid,
-            high,
-            fhigh,
-        )
-
-    flow = func(low)
-    fhigh = func(high)
-    args = (func, low, flow, high, fhigh)
-    return jax.lax.fori_loop(0, niter, _func, args, unroll=15)[-2]
-
-
 # start of code from https://github.com/google/jax/blob/main/jax/_src/numpy/util.py #
 # used with modifications for galsim under the following license:
 # fmt: off