pmocz / advectiondiffusion-jax
Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff
☆11Updated this week
Alternatives and similar repositories for advectiondiffusion-jax:
Users that are interested in advectiondiffusion-jax are comparing it to the libraries listed below
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆29Updated this week
- Finite Volume Constrained Transport MHD simulation of the Orszag-Tang vortex☆19Updated this week
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated last week
- A simple numerical MHD simulation code in python.☆13Updated 2 months ago
- A simple JAX solver for the 2D Euler equations for benchmarking☆20Updated 6 months ago
- Finite difference tools in JAX☆19Updated last year
- 1D-1V Vlasov-Poisson(-Fokker-Planck), Plasma Physics PDE Simulation Tool in NumPy and experiment management in MLFlow☆34Updated last year
- Python package for generating LICs to visualise 2D vector fields☆16Updated this week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆67Updated 2 weeks ago
- Orthogonal polynomials with JAX☆22Updated this week
- New generation astrophysical plasma simulation code with CPU/GPU portability☆36Updated this week
- Efficient Differentiable n-d PDE solvers in JAX.☆28Updated 6 months ago
- Finite Difference simulation of the Wave Equation (Double-slit experiment)☆24Updated this week
- Riemann solver for the Euler equations☆30Updated last year
- Bilinear interpolation on grids with jax☆14Updated 2 years ago
- JAX bindings for the NVIDIA cuDecomp library☆35Updated this week
- Particle In Cell code in Julia☆12Updated 4 months ago
- Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.☆24Updated last year
- Numerical relativity with octree based wavelet adaptive mesh refinement.☆42Updated 2 weeks ago
- Statistics of Turbulence Python Package☆64Updated 4 months ago
- ☆31Updated 4 years ago
- Physics-driven machine learning using PyTorch and Firedrake☆14Updated last year
- Numerical quadrature with JAX☆58Updated this week
- Create Your Own Finite Volume Fluid Simulation (With Python) Part 2: Boundary Conditions and Source Terms☆32Updated this week
- A dispersion relation solver for the hybrid-kinetic model of plasma physics.☆12Updated 6 years ago
- Neural Emulator Architectures in JAX.☆16Updated 6 months ago
- Evolution STEllaire en Rotation.☆20Updated last month
- JAX Tutorial notebooks : basics, crash & tips, usage of optax/JaxOptim/Numpyro☆15Updated 2 months ago
- ☆17Updated 4 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…☆19Updated 4 years ago