Our implementation is designed to perform MD simulations on CPU and ML inference on GPU, with these two processes running in parallel. To achieve optimal performance, we need to carefully balance the workloads on CPU and GPU. For testing, we selected a protein-ligand complex with nearly 42,000 atoms (Figure ). We found that as the number of CPU cores increased up to 16 CPU cores, the simulation performance of both ML models improved.