pmocz / advectiondiffusion-jaxLinks
Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff
☆14Updated 7 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☆75Updated 2 months ago
- Generalized sparse regression for continuous and discrete data☆12Updated 2 weeks ago
- PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.☆57Updated last year
- An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.☆88Updated 5 months ago
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆35Updated last week
- Finite Volume Constrained Transport MHD simulation of the Orszag-Tang vortex☆20Updated 7 months 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…☆22Updated 4 years ago
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆21Updated 8 months ago
- Example problems in Physics informed neural network in JAX☆82Updated 2 years 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
- Introduction to JAX Workshop @ ETH Zurich, 25 June 2024☆39Updated 8 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- ☆55Updated 2 years ago
- A collection of codes and references for deep learning in fluid dynamics☆19Updated 4 years ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆47Updated last year
- Pseudospectral Kolmogorov Flow Solver☆41Updated 2 years ago
- A JAX-based adaptive mesh refinement framework☆27Updated 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
- A Python library for training neural ODEs.☆26Updated 10 months ago
- ☆17Updated last year
- Python interface for libROM, library for reduced order models☆62Updated 4 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆94Updated this week
- LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite☆66Updated 11 months ago
- Create Your Own Navier-Stokes Spectral Method Simulation (With Python)☆53Updated 7 months ago
- This repository contains code, which was used to generate large-scale results in the HINTS paper.☆35Updated last year
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆131Updated last year
- Bilinear interpolation on grids with jax☆16Updated 3 years ago
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆29Updated 2 years ago
- Bootcamp notebooks☆61Updated 4 months ago