3 ms·

I have seen JAX-MD[1] but not sure about "much better". On the other hand, there is just no MD implemented with PyTorch. [1]: https://github.com/jax-md/jax-md

I have seen JAX-MD[1] but not sure about "much better". On the other hand, there is just no MD implemented with PyTorch. [1]: https://github.com/jax-md/jax-md https://github.com/jax-md/jax-md

new_user55 4y ago
There is [torch-md](https://github.com/torchmd/torchmd https://github.com/torchmd/torchmd)