AaltoPML / VAI-Lab
Modular pipeline for design assistance. PYPi Package: https://pypi.org/project/vai-lab
☆13Updated 10 months ago
Related projects ⓘ
Alternatives and complementary repositories for VAI-Lab
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆11Updated 10 months ago
- A generic interface for linear algebra backends☆70Updated 4 months ago
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆21Updated last year
- ☆49Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆95Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆34Updated 2 weeks ago
- Riemannian Optimization Using JAX☆45Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆65Updated 4 years ago
- probabilistic programming focused on fun☆35Updated 2 weeks ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Updated 3 years ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆17Updated last year
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆21Updated 6 months ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆24Updated 5 months ago
- Matrix-free linear algebra in JAX.☆103Updated 2 months ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆38Updated 2 weeks ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 3 years ago
- ☆103Updated 2 weeks ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year
- Supplementary code for the NeurIPS 2020 paper "Matern Gaussian processes on Riemannian manifolds".☆27Updated 3 years ago
- Bayesian algorithm execution (BAX)☆47Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 2 weeks ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆38Updated last year
- Code for Gaussian Score Matching Variational Inference☆27Updated last month
- Simulation-based inference benchmark☆89Updated 7 months ago
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆25Updated last month
- Exponential families for JAX☆55Updated 3 weeks ago
- ☆13Updated 3 weeks ago
- Geometric Dynamic Variational Autoencoders (GD-VAEs) for learning embedding maps for nonlinear dynamics into general latent spaces. This …☆24Updated last year
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆15Updated 3 months ago