Pipeline tests are failing when jax 4.0.34 is used due to TypeErrors. The culprit looks to be in `ScalarRootFind`: https://github.com/sandialabs/optimism/actions/runs/11370570809/job/31630661680#step:5:3167