{"library":"jax-cuda12-plugin","install":[{"cmd":"pip install jax-cuda12-plugin","imports":["import jax","import jax.numpy as jnp"]},{"cmd":"pip install -U --pre jax jaxlib \"jax-cuda12-plugin[with-cuda]\" jax-cuda12-pjrt -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html","imports":[]}]}