jax-md
by jax-md
Jupyter Notebookpushed almost 2 years ago
Differentiable, Hardware Accelerated, Molecular Dynamics
AI summary
MD simulator
A library for simulating molecular dynamics with automatic hardware acceleration and end-to-end differentiability using the JAX framework
- stars
- 1.2K
- forks
- 198
- watching
- 46
- awesome list
- 1
Featured in 1 awesome list
Each link jumps to the spot where the list mentions jax-md.