JAX-SPH

What is JAX-SPH?

JAX-SPH (Toshev et al., 2024) is a Smoothed Particle Hydrodynamics (SPH) code written in JAX. JAX-SPH is designed to be simple, fast, and compatible with deep learning workflows. We currently support the following SPH routines:

• Standard SPH (Adami et al., 2012)

• Transport velocity SPH (Adami et al., 2013)

• Riemann SPH (Zhang et al., 2017)

Check out our GitHub repository for more information including installation instructions and tutorial notebooks.

Getting Started

• Tutorials
• Defaults

API

• Case Setup
  • Case
  • Partition
• Solver
  • SPH Solver
  • Equation of State
  • Kernels
  • Integrator
• Simulate
  • Simulator
• Utils and Defaults
  • Utils
  • Input/Output
  • Defaults