feat[next]: add support for array_namespace allocation, enable jax.numpy testing#2442

havogt wants to merge 4 commits intoGridTools:mainGridTools/gt4py:mainfrom

havogt:jnp_supporthavogt/gt4py:jnp_supportCopy head branch name to clipboard