license:mit size_categories:n<1K format:parquet modality:tabular library:datasets library:dask library:mlcroissant library:polars region:us
---
license: mit
dataset_info:
  features:
  - name: x
    sequence:
      sequence: float32
  - name: y
    sequence:
      sequence: float32
  - name: v_x
    sequence:
      sequence: float32
  - name: v_y
    sequence:
      sequence: float32
  - name: mat
    sequence: int32
  - name: n_particles
    dtype: int64
  - name: n_steps
    dtype: int64
  - name: friction_angle
    dtype: float64
  splits:
  - name: train
    num_bytes: 6494220384
    num_examples: 1000
  download_size: 5836656876
  dataset_size: 6494220384
configs:
- config_name: default
  data_files:
  - split: train
    path: data/train-*
---
## MPM 2D Sand Dataset
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.

## Loading Dataset
```py
from datasets import load_dataset

ds = load_dataset('chhsiao93/mpm_sand_2d')
print(ds['train'][0]) # first trajectory
```