fix(2024Q1): build nlist faster with `torch.amax`

deepmodeling / deepmd-kit

A deep learning package for many-body potential energy representation and molecular dynamics

https://docs.deepmodeling.com/projects/deepmd/

GNU Lesser General Public License v3.0

1.41k stars 487 forks source link

Closed iProzd closed 1 month ago