{"library":"jax-md","install":[{"cmd":"pip install jax-md","imports":["from jax_md import simulate","from jax_md import energy","from jax_md import space","from jax_md import partition"]},{"cmd":"pip install --upgrade jax jaxlib jax-md","imports":[]}]}