JAX behavior at the boundary changed and no longer returns np.inf like scipy does. Fixed by evaluating at a non-boundary value.