py-jaxlib (spack): XLA library for Jax 
Keywords: jax