lammps/ libtorch-gpu/ dataset/ training/ benchmark/ **.ipynb_checkpoints/