DIGIfusion / enchanted-surrogatesLinks
A framework for creating databases for surrogate models of complex physics codes.
☆12Updated this week
Alternatives and similar repositories for enchanted-surrogates
Users that are interested in enchanted-surrogates are comparing it to the libraries listed below
Sorting:
- Numerical quadrature with JAX☆75Updated 3 weeks ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆53Updated this week
- Interpolation and function approximation with JAX☆248Updated last week
- Orthogonal polynomials with JAX☆26Updated 3 weeks ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Updated 2 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated this week
- Neural Emulator Architectures in JAX.☆23Updated last year
- Finite difference tools in JAX☆23Updated 2 years ago
- ☆10Updated 2 months ago
- ☆16Updated last year
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Updated last year
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆115Updated last week
- Matrix-free linear algebra in JAX.☆159Updated 2 months ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆69Updated 2 months ago
- Material for workshop and autumn school on scientific machine learning 2023☆21Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆53Updated 6 months ago
- Prototype for sparse automatic differentiation in JAX + Julia☆20Updated 7 months ago
- Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. ⚡☆28Updated this week
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Updated 3 months ago
- Simulation-based inference in JAX☆45Updated 2 months ago
- Unitful Quantities in JAX☆52Updated last week
- Universal Partial Differential Equations Simulator☆24Updated last year
- 🐈 Differentiable Spectrum Modeling of Exoplanets and Brown Dwarfs☆67Updated 2 weeks ago
- ☆216Updated last week
- Special functions in Jax☆16Updated 11 months ago
- Differentiable Gravitational Waveforms with JAX☆71Updated 6 months ago
- Bilinear interpolation on grids with jax☆16Updated 3 years ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆33Updated last year
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Updated this week