A JAX-powered Material Point Method & Single Integration Point simulation environment for granular materials
HydraxMPM packages the Material Point Method (MPM) solver for large-scale granular dynamics simulations and Single Integration Point (SIP) testing within one environment.
The software is built on the JAX-ecosystem. It leverages automatic differentiation and hardware acceleration (CPU/GPU/TPU) for research and development of numerical models capturing solid-like and fluid-like behavior of granular materials.
-
Install uv: Follow instructions here.
-
Clone & Install Dependencies:
git clone https://github.com/GrainLearning/HydraxMPM.git && cd HydraxMPM # for CPU uv sync --extra cpu # for GPU uv sync --extra
-
Run Example:
uv run ./projects/collapse/collapse.py
- Unified MPM & SIP: Shared API facilitates rapid prototyping and validation.
- High Performance: JAX backend with JIT compilation.
- Differentiable: Enables advanced gradient-based studies.
- Modular: Designed for extensibility in research settings.
- Solvers & Schemes: Explicit MPM (USL) with FLIP/PIC, APIC, AFLIP transfer; Linear, Quadratic, Cubic B-spline basis functions.
-
Available Models: Drucker-Prager, Modified Cam-Clay, Newtonian Fluid, Incompressible
$\mu (I)$ rheology. - SIP Tests: Triaxial (Drained/Undrained), Constant Pressure/Volume Shear, Isotropic Compression.
- Contact & Boundaries: Rigid body contact (penalty-based), slip/no-slip conditions, level-set
- Time Stepping & Stability: Fixed and adaptive time stepping with Courant–Friedrichs–Lewy (CFL) condition.
- Retief Lubbe (Soil Micro Mechanics group / University of Twente)
- Hongyang Cheng (Soil Micro Mechanics group / University of Twente)
This research is part of the project TUSAIL Training in Upscaling Particle Systems: Advancing Industry across Length-scales and has received funding from the European Horizon2020 Framework Programme for research, technological development and demonstration under grant agreement ID 955661.