2022年1月4日
JAX-MD在近邻表的计算中，使用了什么奇技淫巧？（一）
摘要： 本文是第一篇关于JAX-MD的源码学习的文章，主要关注点在于JAX-MD中对于近邻表的检索和优化。本文的主要内容是其中构建CellList的部分，通过打格点的方法可以大大降低近邻表搜索算法的复杂度，在GPU计算的过程中更是可以极大的降低显存的占用，从而允许我们去运行更大规模的体系。 阅读全文
posted @ 2022-01-04 16:11 DECHIN 阅读(741) 评论(0) 推荐(0)