f0uriest / interpax
Interpolation and function approximation with JAX
☆135Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for interpax
- Numerical quadrature with JAX☆44Updated 3 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- ☆107Updated 3 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆200Updated 2 months ago
- A differentiable cosmology library in JAX☆178Updated 3 months ago
- Probabilistic Programming and Nested sampling in JAX☆147Updated last week
- Differentiable and accelerated spherical transforms with JAX☆136Updated this week
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆121Updated 2 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆70Updated this week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆332Updated this week
- 🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt☆57Updated this week
- Stencil computations in JAX☆66Updated last year
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆360Updated last month
- A system for scientific simulation-based inference at scale.☆161Updated 7 months ago
- Stainless neural networks in JAX☆31Updated last month
- Community-sourced list of papers and resources on neural simulation-based inference.☆95Updated 5 months ago
- The tiniest of Gaussian Process libraries☆296Updated last week
- Minimal Implementation of Bayesian Optimization in JAX☆85Updated 6 months ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Generative models with JAX.☆18Updated 10 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 3 weeks ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆52Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆28Updated last week
- Gravitational-wave data analysis tools in Jax☆59Updated 3 weeks ago
- Simulation-based inference in JAX☆21Updated last month