{"library":"jax-md","type":"library","category":null,"description":"JAX-MD is a differentiable, hardware-accelerated molecular dynamics library built on JAX. It supports both rigid and flexible molecules, NVE/NVT/NPT ensembles, and integrates with JAX's autograd for energy minimization and force computation. Current version 0.2.28, requires Python >=3.10. Active development with irregular releases.","language":"python","status":"active","version":"0.2.28","tags":["molecular-dynamics","differentiable","jax","scientific-computing","gpu"],"last_verified":"Sun Jun 07","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":[]}],"homepage":"https://github.com/google/jax-md","github":"https://github.com/google/jax-md","docs":null,"changelog":null,"pypi":null,"npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":null}