{"library":"jax-cuda12-pjrt","install":[{"cmd":"pip install jax jax-cuda12-pjrt","imports":["import jax"]}]}