fix: Access jax.config from jax

[matthewfeickert]

Description

As of jax and jaxlib v0.4.20 accessing jax.config from the jax.config submodule is deprecated and it should be accessed from the jax top level API instead.

>>> from jax.config import config
<stdin>:1: DeprecationWarning: Accessing jax.config via the jax.config submodule is deprecated

This doesn't require any change to the lower bounds of jax and jaxlib

pyhf/pyproject.toml

Lines 76 to 79 in aa22be9

	jax = [
	"jax>=0.4.1", # c.f. PR #2079
	"jaxlib>=0.4.1", # c.f. PR #2079
	]

Checklist Before Requesting Reviewer

• Tests are passing
• "WIP" removed from the title of the pull request
• Selected an Assignee for the PR to be responsible for the log summary

Before Merging

For the PR Assignees:

• Summarize commit messages into a comprehensive review of the PR

* As of jax and jaxlib v0.4.20 accessing jax.config from the jax.config
  submodule is deprecated and it should be accessed from the jax top
  level API instead.

   ```
   >>> from jax.config import config
   <stdin>:1: DeprecationWarning: Accessing jax.config via the jax.config submodule is deprecated
   ```

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (aa22be9) 98.28% compared to head (f029794) 98.28%.

Additional details and impacted files

@@           Coverage Diff           @@
##             main    #2376   +/-   ##
=======================================
  Coverage   98.28%   98.28%           
=======================================
  Files          69       69           
  Lines        4539     4539           
  Branches      803      803           
=======================================
  Hits         4461     4461           
  Misses         45       45           
  Partials       33       33           

Flag	Coverage Δ
contrib	97.86% <100.00%> (ø)
doctest	60.71% <100.00%> (ø)
unittests-3.10	96.29% <100.00%> (ø)
unittests-3.11	96.29% <100.00%> (ø)
unittests-3.8	96.32% <100.00%> (ø)
unittests-3.9	96.34% <100.00%> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

Files	Coverage Δ
src/pyhf/tensor/jax_backend.py	98.61% <100.00%> (ø)

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.