adam-coogan / jaxinterp2d
Bilinear interpolation on grids with jax
☆14Updated 2 years ago
Alternatives and similar repositories for jaxinterp2d:
Users that are interested in jaxinterp2d are comparing it to the libraries listed below
- Numerical quadrature with JAX☆53Updated this week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆55Updated 3 months ago
- Efficient Differentiable n-d PDE solvers in JAX.☆27Updated 4 months ago
- Orthogonal polynomials with JAX☆15Updated this week
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆28Updated last month
- ☆31Updated 4 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆66Updated this week
- N-Point Correlation Functions in Julia☆15Updated 3 years ago
- Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop☆20Updated 9 months ago
- JAX bindings for the NVIDIA cuDecomp library☆32Updated 3 weeks ago
- Special functions in Jax☆12Updated 3 weeks ago
- Differentiable modelling on (multi)-GPUs workshop at JuliaCon23☆44Updated 5 months ago
- Finite difference tools in JAX☆17Updated last year
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆89Updated 3 weeks ago
- 1D-1V Vlasov-Poisson(-Fokker-Planck), Plasma Physics PDE Simulation Tool in NumPy and experiment management in MLFlow☆33Updated last year
- Simulation-based inference in JAX☆31Updated last month
- Differentiable interface to FEniCS for JAX☆53Updated 3 years ago
- Python wrapper of Minpack (root finding) which can be called from within numba functions.☆34Updated 3 years ago
- Two Dimensional Fokker-Planck Solver using Matlab☆19Updated 5 years ago
- Neural Emulator Architectures in JAX.☆16Updated 4 months ago
- Unitful Quantities in JAX☆32Updated last week
- Implementation of Arxiv:2212.08549 in Julia☆12Updated 2 months ago
- Julia routines for working with MHD Equilibrium☆12Updated last year
- A fast, differentiable, and extensible public BBN code☆11Updated 3 weeks ago
- Interpolation and function approximation with JAX☆165Updated this week
- MCHMC: sampler from an arbitrary differentiable distribution☆71Updated 3 weeks ago
- Dynamic dispatch of a list of command-line tasks, locally or on a cluster. Supports retrying failed tasks, and adding/removing compute re…☆45Updated 3 weeks ago
- Solve sparse linear systems in JAX using the KLU algorithm☆34Updated 2 weeks ago
- A small library of paramaterizations and parameter constraints for PyTrees.☆20Updated 5 months ago
- The automatically differentiable and GPU-compatible toolkit for CMB analysis.☆51Updated 5 months ago