Files
lammps/examples/mliap/jax/deploy_script.py
2023-05-20 14:08:20 -06:00

11 lines
262 B
Python

import lammps
import lammps.mliap
#from lammps.mliap.mliap_unified_lj import MLIAPUnifiedLJ
from mliap_unified_jax import MLIAPUnifiedJAX
def create_pickle():
unified = MLIAPUnifiedJAX(["Ar"])
unified.pickle('mliap_unified_jax_Ar.pkl')
create_pickle()