新微比特纳米科技申请基于存算一体架构的机器学习力场高效并行加速方法和系统专利,为大规模、高精度的机器学习分子动力学模拟提供了一种革命性的计算平台

一种基于存算一体架构的机器学习力场高效并行加速方法和系统

  国家知识产权局信息显示，新微比特纳米科技（苏州）有限公司申请一项名为“一种基于存算一体架构的机器学习力场高效并行加速方法和系统”的专利，公开号CN121766481A，申请日期为2025年12月。

  专利摘要显示，本发明公开了一种基于存算一体架构的机器学习力场高效并行加速方法和系统，其特征在于，该方法在由主机处理器和存算一体加速卡组成的异构计算平台上执行，包括以下步骤：

• 在主机处理器端执行：
  • 步骤S101、原子体系数据获取与邻居列表准备；
  • 步骤S102、模型权重映射与配置；
• 在存算一体加速卡端执行：
  • 步骤S103、动态图结构映射；
  • 步骤S104、存内模拟特征更新；
  • 步骤S105、近存数字非线性变换；
  • 步骤S106、混合精度能量聚合；
  • 步骤S107、原位反向传播原子力计算；
• 返回主机处理器端执行：
  • 步骤S108、时间积分与位置更新。

  本发明通过主机与存算一体加速卡的异构协同，以及数据重排与原位计算的创新结合，为大规模、高精度的机器学习分子动力学模拟提供了一种革命性的计算平台。

企业背景

  新微比特纳米科技（苏州）有限公司成立于2021年，位于苏州市，是一家以从事研究和试验发展为主的企业，注册资本1125万人民币。根据公开数据，该公司已参与招投标项目1次，拥有专利信息11条，并取得行政许可2项。