The authors introduce JAX-SPH, a Lagrangian fluid simulator compatible with deep learning frameworks, aiming to enhance the integration of SPH with ML workflows.
