3.8k citations · 7.6k across the 48 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…