kUPS

kUPS 是一个基于 JAX 的高性能原子级模拟工具包，提供可组合、可微分的采样器、势函数和传播器原语。它支持蒙特卡洛（NVT/GCMC）、分子动力学（NVE/NVT/NPT）及几何优化等多种模拟方法，内置 Lennard-Jones、Coulomb、Morse 等经典势及 MACE/UMA 等机器学习力场，并原生支持 CPU/GPU/TPU 加速与自动微分。适用于计算化学、材料科学和物理建模等领域中需大规模并行、可微分或硬件加速的原子尺度模拟任务。