{"library":"jmp","install":[{"cmd":"pip install jmp","imports":["import jmp","from jmp import Policy","from jmp import get_policy","from jmp import DynamicLossScale","from jmp import all_finite"]},{"cmd":"# First, install JAX with accelerator support (e.g., CPU, CUDA, TPU) following JAX's official instructions.\n# Example for CPU:\npip install jax[cpu]\n\npip install jmp","imports":[]}]}