Simulasi dinamika molekul telah mencapai terobosan dengan integrasi potensi interatomik pembelajaran mesin (MLIP) ke dalam paket LAMMPS MD. Kemajuan ini, sebuah kolaborasi antara NVIDIA dan laboratorium nasional, menyederhanakan proses bagi pengembang untuk menghubungkan MLIP ke LAMMPS, meningkatkan skalabilitas dan efisiensi simulasi dalam kimia komputasional dan ilmu material.

Antarmuka ML-IAP-Kokkos yang baru dikembangkan sangat penting untuk memungkinkan simulasi sistem atom yang cepat dan skalabel. Antarmuka ini mendukung model MLIP yang meneruskan pesan dan memanfaatkan kemampuan komunikasi LAMMPS untuk mentransfer data antar GPU secara efisien. Hal ini terutama penting untuk simulasi skala besar yang memanfaatkan daya komputasi beberapa GPU. Dengan menggunakan Cython untuk menjembatani Python dan LAMMPS C++/Kokkos, antarmuka ini memungkinkan akselerasi GPU ujung ke ujung dan mengoptimalkan alur kerja simulasi.

Desain antarmuka ini menghubungkan model PyTorch dengan LAMMPS secara mulus, memfasilitasi simulasi MD yang skalabel. Panduan langkah demi langkah disediakan untuk menyiapkan lingkungan, membangun antarmuka ML-IAP-Kokkos, dan mengimplementasikan dukungan penerusan pesan. Panduan ini ditujukan bagi peneliti yang berpengalaman dengan alat simulasi MD, Python, dan model pembelajaran mesin. Fitur akselerasi opsional juga tersedia untuk model yang mendukung cuEquivariance dan GPU NVIDIA.

Antarmuka ML-IAP-Kokkos merupakan alat penting yang memungkinkan simulasi MD multi-GPU dan multi-node menggunakan MLIP. Antarmuka ini menjembatani kesenjangan antara medan gaya berbasis ML modern dan infrastruktur komputasi berkinerja tinggi, sehingga memungkinkan simulasi sistem yang sangat besar dengan efisiensi tinggi. Tutorial yang disediakan tidak hanya menunjukkan cara menyiapkan dan menjalankan simulasi, tetapi juga mendemonstrasikan manfaat akselerasi GPU dan kemampuan penyampaian pesan dalam LAMMPS. Hasil Benchmark menyoroti potensi antarmuka ini untuk simulasi skala besar, menjadikannya aset yang sangat berharga bagi para peneliti dan pengembang di bidang ini.