kUPS

kUPS 是一个基于 JAX 的高性能原子模拟工具包。它提供了可组合、可微分的基本组件，如采样器、势能和传播器，并支持在 CPU、GPU 和 TPU 上进行硬件加速。项目核心功能包括蒙特卡洛模拟、分子动力学以及几何优化方法，同时支持多种力场与势能模型。kUPS 适合用于需要高效执行大量独立模拟任务的场景，特别是在材料科学、化学及生物物理学研究领域中对复杂系统进行建模和分析时。