PDMD Python package to perform automatic machine-learning (ML) model training, deployment, and assessment for ultrafast high-fidelity molecular dynamics simulations.
Features:
- Support CPU and CUDA GPU.
- Support ML models for the prediction of system energy and atomic forces based on chemically adaptive graph neural network (ChemGNN).
- Support training a unified ML model for multiple chemical systems.