From f029794af2e21b46780228984bbef0fa04b4e29a Mon Sep 17 00:00:00 2001
From: Matthew Feickert <matthew.feickert@cern.ch>
Date: Sat, 4 Nov 2023 01:44:31 -0500
Subject: [PATCH] fix: Access jax.config from jax

* 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
   ```
---
 src/pyhf/tensor/jax_backend.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py
index 6ae53b7bec..61a14084c3 100644
--- a/src/pyhf/tensor/jax_backend.py
+++ b/src/pyhf/tensor/jax_backend.py
@@ -1,4 +1,4 @@
-from jax.config import config
+from jax import config
 
 config.update('jax_enable_x64', True)