pmocz / advectiondiffusion-jaxLinks
Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff
☆14Updated 8 months ago
Alternatives and similar repositories for advectiondiffusion-jax
Users that are interested in advectiondiffusion-jax are comparing it to the libraries listed below
Sorting:
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆77Updated 2 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆35Updated last week
- A JAX-based adaptive mesh refinement framework☆27Updated 5 months ago
- Finite Volume Constrained Transport MHD simulation of the Orszag-Tang vortex☆20Updated 8 months ago
- Generalized sparse regression for continuous and discrete data☆12Updated last month
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆21Updated 8 months ago
- The unsupervised learning problem trains a diffeomorphic spatio-temporal grid, that registers the output sequence of the PDEs onto a non-…☆19Updated 3 years ago
- An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.☆90Updated 5 months ago
- Python interface for libROM, library for reduced order models☆62Updated 4 months ago
- Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"☆39Updated 3 years ago
- Differentiable interface to FEniCS for JAX☆58Updated 4 years ago
- This codes calculates the dimensionalized POD and uses SINDy from the PySINDy python package to build a data-driven model for it. The cod…☆23Updated 4 years ago
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆58Updated last year
- A collection of codes and references for deep learning in fluid dynamics☆19Updated 4 years ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆19Updated 2 months ago
- TorchFSM: Fourier Spectral Method with PyTorch☆53Updated 2 weeks ago
- Pseudospectral Kolmogorov Flow Solver☆41Updated 2 years ago
- Bilinear interpolation on grids with jax☆16Updated 3 years ago
- Riemann solver for the Euler equations☆32Updated 8 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated 2 weeks ago
- Bootcamp notebooks☆62Updated 5 months ago
- Example problems in Physics informed neural network in JAX☆81Updated 2 years ago
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆95Updated 2 months ago
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆66Updated last year
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆39Updated 9 months ago
- Finite difference tools in JAX☆23Updated last year
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆94Updated last week
- JAX bindings for the NVIDIA cuDecomp library☆51Updated 2 months ago
- A Python library for training neural ODEs.☆26Updated 11 months ago