tumaer / jax-sphView external linksLinks
JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
☆77Oct 29, 2025Updated 3 months ago
Alternatives and similar repositories for jax-sph
Users that are interested in jax-sph are comparing it to the libraries listed below
Sorting:
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆68Jan 1, 2025Updated last year
- Neural SPH☆41Nov 23, 2025Updated 2 months ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 2 months ago
- Differentiable Fluid Dynamics Package☆510Jan 29, 2026Updated 2 weeks ago
- [GSI 2023] Learning Lagrangian Fluid Mechanics with E(3)-Equivariant GNNs☆15Jun 3, 2024Updated last year
- Differentiable Finite Element Method with JAX☆582Feb 8, 2026Updated last week
- A fast, differentiable, and extensible public BBN code☆15Jan 28, 2026Updated 2 weeks ago
- ☆43Feb 6, 2026Updated last week
- Code for reproducing the paper "Space-Time Continuous PDE Forecasting using Equivariant Neural Fields" (https://arxiv.org/abs/2406.06660)…☆13Nov 18, 2025Updated 2 months ago
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆41Jan 31, 2026Updated 2 weeks ago
- Implementation of papers in 101 lines of code.☆18Nov 12, 2023Updated 2 years ago
- Bilinear interpolation on grids with jax☆16May 13, 2022Updated 3 years ago
- Deep renormalized Mori-Zwanzig (DrMZ) Julia package.☆17Jun 1, 2023Updated 2 years ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆440Jan 20, 2026Updated 3 weeks ago
- Python 2D Navier-Stokes solver☆27Aug 4, 2025Updated 6 months ago
- Code for the paper Universal Physics Transformers☆149Jul 2, 2025Updated 7 months ago
- Jax like function transformation engine but micro, microjax☆34Oct 25, 2024Updated last year
- Create Your Own Finite Volume Fluid Simulation (With Python) Part 2: Boundary Conditions and Source Terms☆31May 8, 2025Updated 9 months ago
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- Computational Fluid Dynamics in JAX☆920Jan 14, 2026Updated last month
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆40Mar 1, 2025Updated 11 months ago
- Generalizing to New Physical Systems via Context-Informed Dynamics Model☆26May 6, 2024Updated last year
- ☆24Dec 21, 2023Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Jul 15, 2025Updated 7 months ago
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆23Oct 23, 2025Updated 3 months ago
- PyTorch implemention of the Position-induced Transformer for operator learning in partial differential equations☆25Jun 3, 2025Updated 8 months ago
- SPHinXsys provides C++ APIs for engineering simulation and optimization. It aims at complex systems driven by fluid, structure, multi-bod…☆516Updated this week
- ☆48May 4, 2024Updated last year
- Finite Difference simulation of the Wave Equation (Double-slit experiment)☆24May 8, 2025Updated 9 months ago
- A course on developing MPM with Python☆25Jun 11, 2025Updated 8 months ago
- Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141☆25Jan 3, 2024Updated 2 years ago
- Cone program refinement☆10Mar 6, 2020Updated 5 years ago
- JAX implementation of 'A Two-Layer model of Soil Hydrology' (Mahrt & Pan, 1984) for timeseries at a point location.☆10Oct 23, 2024Updated last year
- A fluid simulation system based on the MPS method☆11Feb 10, 2021Updated 5 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 4 months ago
- Julia implementation of LULESH with MPI + X.☆13Jul 23, 2022Updated 3 years ago
- Mayavi workshop material☆10Feb 12, 2022Updated 4 years ago
- Holding opensource codes about research projects related with ML+numerical methods☆11May 11, 2025Updated 9 months ago
- MPM with FEniCS☆10Jun 18, 2019Updated 6 years ago