This dataset includes trajectories of colliding sand cubes simulated using Taichi's MPM framework. The initial position, velocity, size, and number of cubes are randomized.
from datasets import load_dataset
ds = load_dataset('chhsiao93/mpm_sand_2d')
print(ds['train'][0]) # first trajectory
