CederGroupHub / chgnet

Pretrained universal neural network potential for charge-informed atomistic modeling https://chgnet.lbl.gov
https://doi.org/10.1038/s42256-023-00716-3
Other
220 stars 60 forks source link

2.4x avg speedup on PMG struct to CHGNET graph conversion #40

Closed AegisIK closed 1 year ago

AegisIK commented 1 year ago

Converted graph data structure creation (not be confused with the Crystal_Graph object) into c (using Cython as an interfacing layer). Still supports legacy graph creation if c-supported graph creation library import fails.

Overall average graph creation speedup: 3x Overall average Crystal_Graph creation speedup: 2.4x Overall average MD speedup (on GPU): ~2x (may be a noisy benchmark however)

Benchmark on Ac2AgPb supercell:

image