3.6k citations · 10.6k across the 54 of their papers we have counts for
1 paper · 2 filters
Samuel S. Schoenholz, Ekin D. Cubuk
We introduce JAX MD, a software package for performing differentiable physics simulations with a focus on molecular dynamics. JAX MD includes a number of physics simulation environ…