Newer versions of jax introduce breaking change when import jax.config

[fdmalone]

E.g.

OpenFermion/src/openfermion/resource_estimates/thc/utils/thc_objectives.py

Line 6 in e33a29e

from jax.config import config

causes an error for jax > 0.4.23 (which is what the CI is currently pinned to)

[fdmalone]

Need to update as per: jax-ml/jax#18403