Lemuel_/AIMNet2-RXN-NPU
模型介绍
文件和版本
Pull Requests
讨论
分析

isayevlab/aimnet2-rxn 昇腾 NPU 推理适配

模型：isayevlab/aimnet2-rxn（分子力场）

昇腾 NPU 适配结论

项目结果
前向输出CPU/NPU 一致
精度（rel-L2）8.458844867143267e-05（< 1% 阈值，PASS）
性能（NPU avg）5.747651727870107 ms（batch=1）
稳定性20 轮 batch=1 逐位一致（PASS）
验收六件套齐全，CPU/NPU 独立子进程对比

关键适配点

  • aimnet 库 ensemble 加载。
  • CPU/NPU 独立子进程对比，rel-L2 < 1%。

运行方法

python3 minimal_repro.py
python3 accuracy_eval.py
python3 benchmark.py
python3 stability.py
python3 inference.py

结果文件

  • results/accuracy_242.json / benchmark_242.json / stability_242.json / inference_results_242.json

三张真实终端截图

  • assets/agent_workflow.png / assets/npu_device_call.png / assets/model_result.png