convert a pyhf spec in-place to contain jax arrays

jaxify_spec.py

from __future__ import annotations
from jax import numpy as jnp
from typing import Any

def convert_jax(spec: dict[str, Any]) -> None:
    for key, value in spec.items():
        value_type = type(value)
        if value_type == dict:
            convert_jax(value)

        elif value_type == list:
            try:
                v = jnp.asarray(value)
                spec[key] = v
            except TypeError as e:
                assert isinstance(value[0], dict)
                [convert_jax(a) for a in value]
        else:
            continue