HydraxMPM

A JAX-powered Material Point Method & Single Integration Point simulation environment for granular materials

Develop · Diagnose · Simulate

📖 Documentation (coming Soon) | ⚙️ Installation | 🤝 Contributing (coming Soon)

HydraxMPM integrates the Material Point Method (MPM) solver for large-scale granular dynamics simulations and Single Integration Point (SIP) testing , within one environment. Built on JAX, 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.

Capabilities

• 🔬 Diagnose Constitutive Models: Perform controlled SIP tests (triaxial, shear, and compression) for advanced constitutive model analysis.
• ⛰️ Simulate Large-Scale Processes: Model complex, large-deformation granular processes (e.g., landslides) using MPM.
• ⚖️ Validate Across Scales: Compare solid-l ike and fluid-like model behavior at both element (SIP) and system (MPM) levels.
• ∇ Gradient-Aware: Utilize automatic differentiation for sensitivity analysis, inverse problems, and optimization.

Key Features

• 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.
• Time Stepping & Stability: Fixed and adaptive time stepping with Courant–Friedrichs–Lewy (CFL) condition.

Installation

1. Install uv: Follow instructions here.
2. Clone & Install Dependencies:

   git clone https://github.com/GrainLearning/HydraxMPM.git && cd HydraxMPM
   uv sync

3. Run Example:

   uv run ./examples/dambreak/dambreak.py

   (Output in ./examples/dambreak/output)

👥 Contributors:

• Retief Lubbe (Soil Micro Mechanics group / University of Twente)
• Hongyang Cheng (Soil Micro Mechanics group / University of Twente)

🙏 Acknowledgements

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.

Warning

This is a research software under active development (pre-alpha). APIs and functionality are subject to change without notice.