A generic interface for linear algebra backends
☆76Feb 21, 2026Updated last week
Alternatives and similar repositories for lab
Users that are interested in lab are comparing it to the libraries listed below
Sorting:
- Gaussian process modelling in Python☆226Dec 15, 2024Updated last year
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 2 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Feb 10, 2026Updated 2 weeks ago
- Geometric kernels on manifolds, meshes and graphs☆291Feb 22, 2026Updated last week
- A framework for composing Neural Processes in Python☆89Dec 17, 2024Updated last year
- Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX☆23Mar 6, 2023Updated 2 years ago
- Probabilistic inference for models of behaviour☆10Oct 13, 2025Updated 4 months ago
- General framework for Bayesian inversion of continuous hierarchical models☆10Sep 20, 2021Updated 4 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Interactive textbook on state space models☆12Apr 11, 2022Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 7 months ago
- Gaussian processes in JAX and Flax.☆593Feb 20, 2026Updated last week
- Tools for building equivariant polynomials on reductive Lie groups.☆36Jun 29, 2023Updated 2 years ago
- Multiple dispatch in Python☆633Feb 21, 2026Updated last week
- Implementation of the Gaussian Process Autoregressive Regression Model☆70Jan 17, 2025Updated last year
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated last year
- A framework for composing Neural Processes in Julia☆76May 3, 2021Updated 4 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Gaussian Process Factor Analysis with Dynamical Structure☆17Nov 3, 2020Updated 5 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- R&D Python package for processing ICON data with xarray☆10Jul 26, 2023Updated 2 years ago
- Predictive Coding JAX-based library☆89Updated this week
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- ☆13Jan 17, 2024Updated 2 years ago
- ☆10Jul 25, 2022Updated 3 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Gaussian processes with spherical harmonic features in JAX☆15Aug 24, 2025Updated 6 months ago
- Tools for an Aesara-based PPL.☆67Oct 28, 2024Updated last year
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 7 months ago
- A Bayesian optimization toolbox built on TensorFlow☆248Jan 14, 2026Updated last month
- Neural Networks for JAX☆84Sep 24, 2024Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,021Feb 3, 2026Updated 3 weeks ago
- ☆16Nov 24, 2023Updated 2 years ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 2 years ago
- PID-like control implemented as active inference with linear generative models☆11Jul 2, 2020Updated 5 years ago
- A library implementing various linear sparse fitting techniques. Uses Tensorflow for running on GPU.☆14Dec 25, 2016Updated 9 years ago
- Decorator for PyMC3☆50Jun 11, 2021Updated 4 years ago
- Code accompanying the paper "Automating Involutive MCMC using Probabilistic and Differentiable Programming"☆14Jul 21, 2020Updated 5 years ago
- Code for "On the Expressiveness of Approximate Inference in Bayesian Neural Networks"☆13Aug 16, 2021Updated 4 years ago