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