Differentiable, Hardware Accelerated, Molecular Dynamics
Why do you think that https://github.com/patrick-kidger/diffrax is a good alternative to jax-md
Differentiable, Hardware Accelerated, Molecular Dynamics
Why do you think that https://github.com/patrick-kidger/diffrax is a good alternative to jax-md