Simulated datasets for LUFNet-CPC2026

Links to code and LUFnet paper:

1. [LUFnet Code]( https://github.com/Gobliu/LUFNet-CPC2026.git )

2. [Scalable neural network driven molecular dynamics simulation](https://doi.org/10.1016/j.cpc.2026.110036)

The .zip files contain the full dataset used in our LUFnet paper.

Table 1. Dataset summary

The dataset is provided in .pt format, compressed as .zip files, and includes the training and validation datasets for LUFnet, as well as the inference dataset used as input for LUFnet.

The dataset were prepared following the procedure described in the Methods section of the LUFnet paper, using Monte Carlo–generated initial configurations and Velocity-Verlet algorithm.

Training and validation datasets

Due to the large storage requirements, the total training set is split into multiple .pt files and provided in compressed zip format. The training data consist of 13 subsets, which can be merged into a single .pt file to load the full training set. The validation set is stored as a single .pt file.

• Training set (13 files) : Train_set_part*.zip

• Validation set (1 file) : Valid_set.zip

Table 2. Detailed information for Training and validation sets

Table 2 shows detailed information about the .pt files for training and validation sets. Within these files, the trajectories include the initial phase-space configuration by MC simulations and configurations saved every 100 integration steps over a total of 18,000 steps using the Velocity-Verlet. The effective time step for LUFnet can be adjusted to τ=0.1.

After merging the separate training subsets into a single .pt file, the loaded tensor for the training set has the shape [nsamples, 3, time points, nparticles, dim] = [180,000, 3, 181, 64, 3]. While the validation dataset has the shape [20,000, 3, 181, 64, 3].

In this study, phase-space points were read off at larger time intervals (τ=0.05 for the 3D LJ system) and used as the training labels.

Inference dataset

The inference data are prepared as the sequence of input for LUFnet, similar to training data generated from the initial configurations. The inference datasets cover different conditions, including various numbers of particles, n = 64, 128, 256. 

• Input for LUFnet (3 folders inside zip) : Inference_input_LUFnet.zip

Table 3. Detailed information for inference set 

Table 3 shows detailed information about the .pt files for inference set. Within these files, the trajectories include the initial phase-space configuration by MC simulations and configurations saved every 100 integration steps over a total of 8,000 steps using the Velocity-Verlet. The effective time step for LUFnet can be adjusted to τ=0.1.

The loaded tensor for input for LUFnet has shape [1000, 3, 81, nparticles, 3] for each condition. 

Rollout data for the Velocity-Verlet algorithm for comparison with LUFnet

Simulations were performed using both LUFnet and Velocity-Verlet algorithm up to t= 100. For comparison with LUFnet (time step of = 0.05), the Velocity Verlet starts at t = 0.35. The configuration at t=0.35 is taken from  the LUFnet input sequence and defined as Index 0. A time step of τ = 0.001 was used with a total of 100,000 rollout steps; these data are provided here.  

• Rollout steps for Velocity-Verlet (3 folders inside zip) : Inference_Rollout_steps_Velocity_verlet.zip

Table 4. Detailed information for rollout steps for Velocity-Verlet (to compare with LUFnet)

Table 4 shows detailed information about the .pt files for rollout steps for Velocity-Verlet. Within these files, 

the loaded tensor for Velocity Verlet rollout data has shape [1000, 3, 1001, nparticles, 3] for each condition. 

For LUFnet, the time integration step was set to τ = 0.05, corresponding to 2,000 rollout steps in the Implementation provided on our github repository.

The 1,000 inference samples are split into 5 groups of 200 samples each for RDF and energy calculations. Metrics are computed for each group, and the mean and standard deviation across the 5 groups are used for plotting and table, as shown in Figures 3 and 4, and Table 2 of the LUFnet paper.