This was observed on a recent PR #567
- The timing is slower on JAX version (GitHub actions build -
gpu) than the numba version. (LHS = this PR, RHS = Live site)~~ (Status: Local testing shows jax should run approximately 2 x faster than numba version, so this timing is an issue with GA environment).
We should investigate performance comparing GitHub Actions to Local and understand why.