https://github.com/jax-md/jax-md
Jupyter Notebook1056
3 months ago
Differentiable, Hardware Accelerated, Molecular Dynamics
Apache License 2.0