diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py index f7eef3aff2..4ea2509c71 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)